| /external/tensorflow/tensorflow/core/kernels/ |
| D | dynamic_stitch_op_gpu.cu.cc | 34 GpuDeviceArrayStruct<int32> input_indices, in DynamicStitchKernel()
54 const GpuDeviceArrayStruct<int>& input_indices, in DynamicStitchGPUImpl()
|
| D | sparse_split_op.cc | 34 const Tensor& input_indices = context->input(1); in Compute() local
|
| D | sparse_slice_op.cc | 31 const Tensor& input_indices = context->input(0); in Compute() local
|
| 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 | 57 const Tensor* input_indices; in Compute() local
339 const Tensor* input_indices; in Compute() local
|
| D | aggregate_ops.cc | 54 gtl::InlinedVector<int, 8> input_indices(num); in Compute() local
|
| D | sparse_tensors_map_ops.cc | 170 const Tensor* input_indices; in Compute() local
225 const Tensor* input_indices; in Compute() local
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/ |
| D | quantization_util.cc | 90 TfLiteContext* context, const std::vector<uint32_t>& input_indices, in DequantizeInputs()
99 TfLiteContext* context, const std::vector<int64_t>& input_indices, in DequantizeInputs()
|
| D | quantization_util_test.cc | 163 std::vector<uint32_t> input_indices = {1}; in TEST() local
188 std::vector<int64_t> input_indices = {1}; in TEST() local
|
| /external/tensorflow/tensorflow/lite/toco/graph_transformations/ |
| D | reorder_reshape_transpose.cc | 62 std::vector<int> input_indices; in ComputeNewPerm() local
|
| D | resolve_constant_unary.cc | 66 std::vector<int> input_indices = ReverseOffset(input_shape, input_offset); in ReduceGeneric() local
|
| /external/tensorflow/tensorflow/java/src/main/native/ |
| D | graph_jni.cc | 214 jintArray input_indices = env->NewIntArray(ninputs); in buildSubgraph() local
258 jintArray input_indices, jstring name, jobject cond_graph_builder, in Java_org_tensorflow_Graph_whileLoop()
|
| /external/libtextclassifier/native/annotator/ |
| D | annotator_jni.cc | 633 const CodepointSpan input_indices = in TC3_JNI_METHOD() local
664 const CodepointSpan input_indices = in TC3_JNI_METHOD() local
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | sparse_ops.cc | 609 ShapeHandle input_indices = c->input(0); in __anon36a8fc881802() local
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/experimental/conv_emitter/ |
| D | conv_emitter.cc | 295 std::vector<mlir::AffineExpr> input_indices; in CreateNaiveMlirConv() local
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | literal.h | 1135 absl::Span<const int64> input_indices = output_indices; in Replicate() local
|
| /external/tensorflow/tensorflow/python/framework/ |
| D | python_op_gen.cc | 226 const std::vector<int>* input_indices, in FlattenInputs()
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/ |
| D | flatbuffer_import.cc | 814 const tflite::SubGraphT& subgraph, ArrayRef<int32_t> input_indices, in PruneSubgraph()
|
| /external/tensorflow/tensorflow/lite/core/ |
| D | subgraph.cc | 610 TfLiteStatus Subgraph::CheckInputAndOutputForOverlap(const int* input_indices, in CheckInputAndOutputForOverlap()
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/ir/ |
| D | tfl_ops.cc | 2544 int output_axis, std::vector<uint64_t> *input_indices, in ComputePermutation()
2599 std::vector<uint64_t> input_indices(num_dimensions); in fold() local
|
| /external/tensorflow/tensorflow/python/ops/parallel_for/ |
| D | pfor.py | 1741 def _inputs_with_flattening(pfor_input, input_indices): argument
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | legalize_tf.cc | 5212 SmallVector<Value, 6> input_indices; in matchAndRewrite() local
|