| /external/libtextclassifier/native/utils/tflite/ |
| D | dist_diversification.cc | 83 TfLiteTensor& output_indices = in AllocateOutputIndexes() local
97 TfLiteTensor& output_indices = in Prepare() local
110 TfLiteTensor& output_indices = in Eval() local
|
| /external/ComputeLibrary/src/core/CPP/kernels/ |
| D | CPPNonMaximumSuppressionKernel.cpp | 38 …ments(const ITensorInfo *bboxes, const ITensorInfo *scores, const ITensorInfo *output_indices, uns… in validate_arguments()
62 …ernel::configure(const ITensor *input_bboxes, const ITensor *input_scores, ITensor *output_indices, in configure()
85 …::validate(const ITensorInfo *bboxes, const ITensorInfo *scores, const ITensorInfo *output_indices, in validate()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | deserialize_sparse_string_op.cc | 82 Tensor output_indices; in Compute() local
105 Tensor output_indices; in Compute() local
232 Tensor* output_indices, Tensor* output_values, Tensor* output_shape) { in GetAndValidateSparseTensor()
|
| D | deserialize_sparse_variant_op.cc | 59 const Tensor* output_indices; in Compute() local
152 Tensor* output_indices; in Compute() local
304 int index, int expected_rank, const Tensor** output_indices, in GetAndValidateSparseTensorIndicesAndValues()
|
| D | sparse_slice_op_gpu.cu.cc | 94 int64_t* __restrict__ output_indices, T* __restrict__ output_values) { in SparseSliceGatherKernel()
152 Tensor* output_indices = nullptr; in operator ()() local
216 Tensor* output_indices = nullptr; in operator ()() local
|
| D | sparse_fill_empty_rows_op_gpu.cu.cc | 132 const Tindex* num_new_rows_before, Tindex* output_indices, T* output_values, in ScatterInputElementsKernel()
154 const bool* empty_row_indicator, Tindex* output_indices, T* output_values) { in ScatterNewElementsKernel()
190 Tindex* output_indices; in operator ()() local
374 Tindex* output_indices; in operator ()() local
|
| D | sparse_slice_grad_op.cc | 80 const Tensor *backprop_val_grad, *input_indices, *output_indices, *input_start; in Compute() local
|
| D | reshape_util_gpu.cu.cc | 35 Tindex* __restrict__ output_indices) { in ReshapeSparseTensorKernel()
|
| D | serialize_sparse_op.cc | 195 Tensor output_indices = Tensor(DT_INT64, {num_entries, rank - 1}); in operator ()() local
296 Tensor& output_indices = serialized_sparse_t(b, 0).emplace<Tensor>( in operator ()() local
|
| D | sparse_split_op_gpu.cu.cc | 168 Index* __restrict__* __restrict__ output_indices = in SparseSplitScatterKernel() local
313 GpuDeviceArrayOnHost<Index*> output_indices(context, num_split); in operator ()() local
|
| D | sparse_tensors_map_ops.cc | 307 Tensor output_indices = Tensor(DT_INT64, {num_entries, rank - 1}); in Compute() local
399 const Tensor& output_indices = st.indices(); in Compute() local
|
| D | sparse_slice_grad_op_gpu.cu.cc | 97 MultiIndexSearchIterator output_indices, in SparseSliceGradKernel()
|
| D | count_ops.cc | 58 auto output_indices = indices->matrix<int64_t>(); in OutputSparse() local
|
| D | sparse_fill_empty_rows_op.cc | 156 auto output_indices = output_indices_t->matrix<Tindex>(); in operator ()() local
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/ |
| D | quantization_util.cc | 108 TfLiteContext* context, const std::vector<uint32_t>& output_indices, in QuantizeOutputs()
118 TfLiteContext* context, const std::vector<int64_t>& output_indices, in QuantizeOutputs()
|
| D | quantization_util_test.cc | 211 std::vector<uint32_t> output_indices = {0}; in TEST() local
233 std::vector<int64_t> output_indices = {0}; in TEST() local
|
| D | operations.h | 178 bool output_indices = false; member
190 bool output_indices = false; member
|
| /external/tensorflow/tensorflow/core/kernels/image/ |
| D | non_max_suppression_op.cu.cc | 328 Tensor* output_indices = nullptr; in DoNMS() local
402 Tensor* output_indices = nullptr; in DoNMS() local
427 Tensor* output_indices = nullptr; in DoNMS() local
546 Tensor* output_indices = nullptr; in Compute() local
592 Tensor* output_indices = nullptr; in Compute() local
643 Tensor* output_indices = nullptr; in Compute() local
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | sparse_ops.cc | 331 ShapeHandle output_indices = in __anonc55240e60e02() local
374 ShapeHandle output_indices = in __anonc55240e61002() local
628 ShapeHandle output_indices = in __anonc55240e61802() local
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/ |
| D | pooling.cc | 130 bool output_indices, GPUOperation* op) { in GetMaxPoolingKernelCode()
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | reverse_test.cc | 93 std::vector<int64_t> output_indices(spec.input_dims.size()); in TEST_P() local
|
| /external/tensorflow/tensorflow/examples/custom_ops_doc/multiplex_3/ |
| D | multiplex_3_kernel.cc | 152 auto output_indices = output_indices_tensor->matrix<int64_t>(); in Compute() local
|
| /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
|
| D | resolve_constant_unary.cc | 66 std::vector<int> output_indices(input_shape.dimensions_count()); in ReduceGeneric() local
|
| /external/tensorflow/tensorflow/java/src/main/native/ |
| D | graph_jni.cc | 216 jintArray output_indices = env->NewIntArray(noutputs); in buildSubgraph() local
|