| /external/tensorflow/tensorflow/core/kernels/ |
| D | sparse_slice_grad_op.cc | 30 const Tensor *backprop_val_grad, *input_indices, *output_indices, *input_start; in Compute() local
34 OP_REQUIRES_OK(ctx, ctx->input("output_indices", &output_indices)); in Compute()
38 TensorShapeUtils::IsMatrix(output_indices->shape()), in Compute()
43 output_indices->shape().DebugString())); in Compute()
51 input_indices->dim_size(1) == output_indices->dim_size(1), in Compute()
54 output_indices->dim_size(1))); in Compute()
56 ctx, output_indices->dim_size(0) <= input_indices->dim_size(0), in Compute()
59 output_indices->dim_size(0), " and ", in Compute()
62 ctx, backprop_val_grad->NumElements() == output_indices->dim_size(0), in Compute()
66 output_indices->dim_size(0))); in Compute()
[all …]
|
| D | deserialize_sparse_variant_op.cc | 59 const Tensor* output_indices; in Compute() local
69 output_shape->NumElements(), &output_indices, in Compute()
71 context->set_output(0, *output_indices); in Compute()
152 Tensor* output_indices; in Compute() local
159 &output_indices)); in Compute()
169 int64* output_indices_data = output_indices->matrix<int64>().data(); in Compute()
304 int index, int expected_rank, const Tensor** output_indices, in GetAndValidateSparseTensorIndicesAndValues() argument
307 *output_indices = serialized_indices.get<Tensor>(); in GetAndValidateSparseTensorIndicesAndValues()
308 if (*output_indices == nullptr) { in GetAndValidateSparseTensorIndicesAndValues()
312 if ((*output_indices)->dtype() != DT_INT64) { in GetAndValidateSparseTensorIndicesAndValues()
[all …]
|
| D | deserialize_sparse_string_op.cc | 82 Tensor output_indices; in Compute() local
89 &output_indices, &output_values, &output_shape)); in Compute()
90 context->set_output(0, output_indices); in Compute()
105 Tensor output_indices; in Compute() local
111 serialized_sparse_t(i, 2), dtype_, i, &output_indices, in Compute()
113 int64 num_entries = output_indices.dim_size(0); in Compute()
114 int rank = output_indices.dim_size(1); in Compute()
119 const auto& output_indices_t = output_indices.matrix<int64>(); in Compute()
232 Tensor* output_indices, Tensor* output_values, Tensor* output_shape) { in GetAndValidateSparseTensor() argument
234 TF_RETURN_IF_ERROR(this->Deserialize(serialized_indices, output_indices)); in GetAndValidateSparseTensor()
[all …]
|
| D | sparse_tensors_map_ops.cc | 296 Tensor output_indices = Tensor(DT_INT64, {num_entries, rank - 1}); in Compute() local
299 auto output_indices_t = output_indices.matrix<int64>(); in Compute()
311 SparseTensor::Create(output_indices, output_values, in Compute()
388 const Tensor& output_indices = st.indices(); in Compute() local
392 OP_REQUIRES(context, TensorShapeUtils::IsMatrix(output_indices.shape()), in Compute()
396 output_indices.shape().DebugString())); in Compute()
409 int64 num_entries = output_indices.dim_size(0); in Compute()
416 int rank = output_indices.dim_size(1); in Compute()
428 DT_INT64, TensorShape({num_entries, 1 + output_indices.dim_size(1)})); in Compute()
430 const auto& output_indices_t = output_indices.matrix<int64>(); in Compute()
|
| D | serialize_sparse_op.cc | 193 Tensor output_indices = Tensor(DT_INT64, {num_entries, rank - 1}); in operator ()() local
196 auto output_indices_t = output_indices.matrix<int64>(); in operator ()()
206 serialize(output_indices, &serialized_sparse_t(b, 0)); in operator ()()
292 Tensor& output_indices = serialized_sparse_t(b, 0).emplace<Tensor>( in operator ()() local
298 static_cast<int64*>(DMAHelper::base(&output_indices)); in operator ()()
|
| D | sparse_fill_empty_rows_op.cc | 143 auto output_indices = output_indices_t->matrix<Tindex>(); in operator ()() local
158 std::copy_n(&indices(i, 0), rank, &output_indices(output_i, 0)); in operator ()()
173 output_indices(starting_index, 0) = row; in operator ()()
175 output_indices(starting_index, col) = 0; in operator ()()
|
| /external/libtextclassifier/native/utils/tflite/ |
| D | dist_diversification.cc | 83 TfLiteTensor& output_indices = in AllocateOutputIndexes() local
86 return context->ResizeTensor(context, &output_indices, in AllocateOutputIndexes()
97 TfLiteTensor& output_indices = in Prepare() local
100 tflite::SetTensorToDynamic(&output_indices); in Prepare()
110 TfLiteTensor& output_indices = in Eval() local
113 if (tflite::IsDynamicTensor(&output_indices)) { in Eval()
133 std::copy(indices.begin(), indices.end(), output_indices.data.i32); in Eval()
134 std::fill_n(output_indices.data.i32 + indices.size(), in Eval()
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | reverse_test.cc | 94 std::vector<int64> output_indices(spec.input_dims.size()); in TEST_P() local
97 output_indices[i] = indices[i]; in TEST_P()
101 output_indices[dim] = (spec.input_dims[dim] - 1) - indices[dim]; in TEST_P()
103 expected.Set<float>(output_indices, value); in TEST_P()
|
| /external/tensorflow/tensorflow/core/kernels/image/ |
| D | non_max_suppression_op.cu.cc | 421 Tensor* output_indices = nullptr; in DoNMS() local
423 context->allocate_output(0, TensorShape({0}), &output_indices)); in DoNMS()
494 Tensor* output_indices = nullptr; in DoNMS() local
500 &output_indices)); in DoNMS()
519 Tensor* output_indices = nullptr; in DoNMS() local
523 0, TensorShape({max_output_size}), &output_indices)); in DoNMS()
528 output_indices->flat<int>().data())); in DoNMS()
532 &output_indices)); in DoNMS()
543 (*output_indices).flat<int>().data())); in DoNMS()
617 Tensor* output_indices = nullptr; in Compute() local
[all …]
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/ |
| D | quantization_util.cc | 108 TfLiteContext* context, const std::vector<uint32_t>& output_indices, in QuantizeOutputs() argument
110 for (auto index : output_indices) { in QuantizeOutputs()
118 TfLiteContext* context, const std::vector<int64_t>& output_indices, in QuantizeOutputs() argument
120 for (auto index : output_indices) { in QuantizeOutputs()
|
| D | quantization_util.h | 49 TfLiteContext* context, const std::vector<uint32_t>& output_indices,
53 TfLiteContext* context, const std::vector<int64_t>& output_indices,
|
| D | quantization_util_test.cc | 211 std::vector<uint32_t> output_indices = {0}; in TEST() local
214 auto status = QuantizeOutputs(&context, output_indices, quant_conversion_map); in TEST()
233 std::vector<int64_t> output_indices = {0}; in TEST() local
236 auto status = QuantizeOutputs(&context, output_indices, quant_conversion_map); in TEST()
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/ |
| D | pooling.cc | 146 bool stride_correction, bool output_indices, in GetMaxPoolingKernelCode() argument
158 if (output_indices) { in GetMaxPoolingKernelCode()
211 if (output_indices) { in GetMaxPoolingKernelCode()
245 if (output_indices) { in GetMaxPoolingKernelCode()
279 if (output_indices) { in GetMaxPoolingKernelCode()
303 attr.output_indices, &op); in CreatePooling()
327 attr.output_indices, &op); in CreatePooling()
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/ |
| D | SparseSparseMaximum.pbtxt | 28 name: "output_indices"
80 name: "output_indices"
134 name: "output_indices"
189 name: "output_indices"
|
| D | SparseReduceMaxSparse.pbtxt | 20 name: "output_indices"
75 name: "output_indices"
132 name: "output_indices"
190 name: "output_indices"
|
| D | SparseReduceSumSparse.pbtxt | 20 name: "output_indices"
80 name: "output_indices"
142 name: "output_indices"
205 name: "output_indices"
|
| D | SparseSparseMinimum.pbtxt | 28 name: "output_indices"
85 name: "output_indices"
144 name: "output_indices"
204 name: "output_indices"
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/ |
| D | SparseReduceMaxSparse.pbtxt | 20 name: "output_indices"
75 name: "output_indices"
132 name: "output_indices"
190 name: "output_indices"
|
| D | SparseSparseMaximum.pbtxt | 28 name: "output_indices"
80 name: "output_indices"
134 name: "output_indices"
189 name: "output_indices"
|
| D | SparseReduceSumSparse.pbtxt | 20 name: "output_indices"
80 name: "output_indices"
142 name: "output_indices"
205 name: "output_indices"
|
| D | SparseSparseMinimum.pbtxt | 28 name: "output_indices"
85 name: "output_indices"
144 name: "output_indices"
204 name: "output_indices"
|
| /external/tensorflow/tensorflow/core/util/sparse/ |
| D | sparse_tensor.h | 498 std::vector<Tensor> output_indices; in Split() local
501 output_indices.reserve(num_split); in Split()
535 output_indices.emplace_back(DT_INT64, in Split()
540 output_indices_t.emplace_back(output_indices[i].matrix<int64>()); in Split()
566 Create(output_indices[i], output_values[i], output_shapes[i], &tensor); in Split()
629 Tensor output_indices(DT_INT64, TensorShape({count, dims})); in Slice()
632 auto output_indices_t = output_indices.matrix<int64>(); in Slice()
659 return SparseTensor(output_indices, output_values, output_shape); in Slice()
|
| /external/tensorflow/tensorflow/lite/toco/graph_transformations/ |
| D | resolve_constant_binary.cc | 127 const std::vector<int> output_indices = ReverseOffset(output_shape, k); in EvaluateBinaryOperatorOnConstantInputs() local
129 modulo_indices[i] = output_indices[i] % small_sizes[i]; in EvaluateBinaryOperatorOnConstantInputs()
131 PairwiseVectorSelect(input0_larger, output_indices, modulo_indices, in EvaluateBinaryOperatorOnConstantInputs()
164 output_data[Offset(output_shape, output_indices)] = outval; in EvaluateBinaryOperatorOnConstantInputs()
|
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_SparseReorder.pbtxt | 23 name: "output_indices"
32 1-D. `N` non-empty values corresponding to `output_indices`.
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | sparse_ops.py | 1164 output_indices, output_values, output_shape = gen_sparse_ops.sparse_slice(
1172 return sparse_tensor.SparseTensor(output_indices, output_values,
2153 (output_indices, output_values, empty_row_indicator,
2160 indices=output_indices,
2335 output_indices, output_values, output_shape = (
2339 output_indices.set_shape([None, rank])
2342 return sparse_tensor.SparseTensor(output_indices, output_values, output_shape)
2408 output_indices, output_values, output_shape = (
2413 output_indices.set_shape([None, rank])
2416 return sparse_tensor.SparseTensor(output_indices, output_values, output_shape)
[all …]
|