| /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/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_grad_op.cc | 30 const Tensor *backprop_val_grad, *input_indices, *output_indices, *input_start; in Compute() local
|
| D | serialize_sparse_op.cc | 193 Tensor output_indices = Tensor(DT_INT64, {num_entries, rank - 1}); in operator ()() local
292 Tensor& output_indices = serialized_sparse_t(b, 0).emplace<Tensor>( in operator ()() local
|
| D | sparse_tensors_map_ops.cc | 296 Tensor output_indices = Tensor(DT_INT64, {num_entries, rank - 1}); in Compute() local
388 const Tensor& output_indices = st.indices(); in Compute() local
|
| D | sparse_fill_empty_rows_op.cc | 143 auto output_indices = output_indices_t->matrix<Tindex>(); in operator ()() local
|
| D | count_ops.cc | 49 auto output_indices = indices->matrix<int64>(); in OutputSparse() local
|
| D | decode_proto_op.cc | 684 std::vector<int> output_indices; in DecodeProtoOp() 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 | 164 bool output_indices = false; member
176 bool output_indices = false; member
|
| /external/tensorflow/tensorflow/core/kernels/image/ |
| D | non_max_suppression_op.cu.cc | 421 Tensor* output_indices = nullptr; in DoNMS() local
494 Tensor* output_indices = nullptr; in DoNMS() local
519 Tensor* output_indices = nullptr; in DoNMS() local
617 Tensor* output_indices = nullptr; in Compute() local
663 Tensor* output_indices = nullptr; in Compute() local
714 Tensor* output_indices = nullptr; in Compute() local
|
| D | non_max_suppression_op.cc | 279 Tensor* output_indices = nullptr; in DoNonMaxSuppressionOp() local
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | sparse_ops.cc | 327 ShapeHandle output_indices = in __anon36a8fc880e02() local
370 ShapeHandle output_indices = in __anon36a8fc881002() local
622 ShapeHandle output_indices = in __anon36a8fc881802() local
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/ |
| D | pooling.cc | 146 bool stride_correction, bool output_indices, in GetMaxPoolingKernelCode()
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | reverse_test.cc | 94 std::vector<int64> output_indices(spec.input_dims.size()); in TEST_P() 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 | 64 std::vector<int> output_indices(input_shape.dimensions_count()); in ReduceGeneric() local
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | unique_op.cc | 250 auto output_indices = xla::GetTupleElement(outer_while, 1); in Compile() local
|
| /external/tensorflow/tensorflow/java/src/main/native/ |
| D | graph_jni.cc | 216 jintArray output_indices = env->NewIntArray(noutputs); in buildSubgraph() local
|
| /external/tensorflow/tensorflow/core/util/sparse/ |
| D | sparse_tensor.h | 498 std::vector<Tensor> output_indices; in Split() local
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/ |
| D | flatbuffer_import.cc | 815 ArrayRef<int32_t> output_indices) { in PruneSubgraph()
|
| /external/tensorflow/tensorflow/lite/core/ |
| D | subgraph.cc | 612 const int* output_indices, in CheckInputAndOutputForOverlap()
|