| /external/libtextclassifier/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_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 | deserialize_sparse_string_op.cc | 80 Tensor output_indices; in Compute() local
103 Tensor output_indices; in Compute() local
229 Tensor* output_indices, Tensor* output_values, Tensor* output_shape) { in GetAndValidateSparseTensor()
|
| D | sparse_slice_grad_op.cc | 30 const Tensor *backprop_val_grad, *input_indices, *output_indices, *input_start; in Compute() local
|
| D | sparse_tensors_map_ops.cc | 295 Tensor output_indices = Tensor(DT_INT64, {num_entries, rank - 1}); in Compute() local
387 const Tensor& output_indices = st.indices(); in Compute() local
|
| D | sparse_fill_empty_rows_op.cc | 145 auto output_indices = output_indices_t->matrix<int64>(); in Compute() local
|
| D | serialize_sparse_op.cc | 240 Tensor output_indices = Tensor(DT_INT64, {num_entries, rank - 1}); in Compute() local
|
| D | decode_proto_op.cc | 681 std::vector<int> output_indices; in DecodeProtoOp() local
|
| D | non_max_suppression_op.cc | 218 Tensor* output_indices = nullptr; in DoNonMaxSuppressionOp() local
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | sparse_ops.cc | 287 ShapeHandle output_indices = in __anone6e195410c02() local
330 ShapeHandle output_indices = in __anone6e195410e02() local
582 ShapeHandle output_indices = in __anone6e195411602() local
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | reverse_test.cc | 93 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 | 126 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/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 | 575 std::vector<Tensor> output_indices; in Split() local
|