| /external/libtextclassifier/native/utils/ |
| D | tflite-model-executor.h | 79 void SetInput(const int input_index, const TensorView<T>& input_data, in SetInput()
86 void SetInput(const int input_index, const std::vector<T>& input_data, in SetInput()
93 void SetInput(const int input_index, const T input_value, in SetInput()
|
| /external/tensorflow/tensorflow/lite/testing/nnapi_tflite_zip_tests/ |
| D | parse_testdata.cc | 137 for (int input_index = 0; input_index < inputs; input_index++) { in ParseExamples() local
145 for (int input_index = 0; input_index < outputs; input_index++) { in ParseExamples() local
158 int input_index = interpreter->inputs()[i]; in FeedExample() local
166 int input_index = interpreter->inputs()[i]; in FeedExample() local
|
| /external/tensorflow/tensorflow/lite/testing/ |
| D | parse_testdata.cc | 134 for (int input_index = 0; input_index < inputs; input_index++) { in ParseExamples() local
142 for (int input_index = 0; input_index < outputs; input_index++) { in ParseExamples() local
155 int input_index = interpreter->inputs()[i]; in FeedExample() local
163 int input_index = interpreter->inputs()[i]; in FeedExample() local
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | xla_helpers.h | 115 int input_index; member
124 int input_index; member
|
| D | tf2xla.cc | 118 int64_t input_index = xla_args.size() - config.variable_size(); in ConvertGraphToXla() local
|
| /external/XNNPACK/test/ |
| D | depth-to-space-operator-tester.h | 175 const size_t input_index = in TestNHWCxX32() local
241 const size_t input_index = in TestNCHW2NHWCxX32() local
|
| D | constant-pad-operator-tester.h | 163 const size_t input_index = in TestX8() local
278 const size_t input_index = in TestX16() local
393 const size_t input_index = in TestX32() local
|
| D | depthtospace-microkernel-tester.h | 129 const size_t input_index = in Test() local
|
| /external/tensorflow/tensorflow/lite/toco/graph_transformations/ |
| D | propagate_fake_quant_num_bits.cc | 124 bool DoesOpInputBlockBackwardPropagation(const Operator& op, int input_index) { in DoesOpInputBlockBackwardPropagation()
152 for (size_t input_index = 0; input_index < op->inputs.size(); ++input_index) { in RecursivelyBackwardPropagateDataType() local
|
| D | remove_trivial_passthrough.cc | 72 int input_index) { in RemoveTrivialPassthroughOp()
|
| D | quantize.cc | 226 std::size_t input_index, ArrayDataType* quantized_data_type, in ChooseQuantizationForOperatorInput()
574 for (std::size_t input_index = 0; input_index < op.inputs.size(); in Run() local
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | dynamic_stitch_op_gpu.cu.cc | 42 const int32 input_index = data_indices[slice_id]; in DynamicStitchKernel() local
|
| D | reshape_util_gpu.cu.cc | 37 const Tindex* input_index = &input_indices[sparse_index * input_rank]; in ReshapeSparseTensorKernel() local
|
| D | quantized_batch_norm_op.cc | 55 const int input_index = (row_index * depth) + channel; in ReferenceBatchNorm() local
148 const int input_index = (row_index * depth) + channel; in FixedPointBatchNorm() local
|
| /external/tensorflow/tensorflow/c/ |
| D | kernels_experimental.cc | 160 void TF_AssignVariable(TF_OpKernelContext* ctx, int input_index, in TF_AssignVariable()
197 void TF_AssignUpdateVariable(TF_OpKernelContext* ctx, int input_index, in TF_AssignUpdateVariable()
327 int32_t input_index, in TF_OpKernelContext_ForwardRefInputToRefOutput()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| D | space_to_depth.h | 66 const int input_index = Offset(input_shape, in_b, in_h, in_w, in_d); in SpaceToDepth() local
|
| D | depth_to_space.h | 65 const int input_index = Offset(input_shape, in_b, in_h, in_w, in_d); in DepthToSpace() local
|
| D | prelu.h | 46 int input_index = SubscriptToIndex(desc1, b, y, x, c); in BroadcastPrelu4DSlow() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | optimize_input_output_buffer_alias.cc | 74 const ShapeIndex& input_index = entry.index; in Build() local
|
| /external/tensorflow/tensorflow/lite/c/ |
| D | c_api.cc | 150 const TfLiteInterpreter* interpreter, int32_t input_index) { in TfLiteInterpreterGetInputTensor()
155 int32_t input_index, in TfLiteInterpreterResizeInputTensor()
|
| D | c_api_experimental.cc | 94 const TfLiteInterpreter* interpreter, int32_t input_index) { in TfLiteInterpreterGetInputTensorIndex()
|
| /external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/transforms/ |
| D | lhlo_elemental_utils.cc | 124 SmallVector<Value, 4> input_index; in elementalLower() local
160 SmallVector<Value, 4> input_index; in elementalLowerImplForBroadcastInDimOps() local
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | mark_input_output_aliases.cc | 38 int input_index; member
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/ |
| D | quantization_util.cc | 33 TfLiteContext* context, int input_index, in DequantizeInput()
|
| /external/tensorflow/tensorflow/lite/experimental/examples/unity/TensorFlowLitePlugin/Assets/TensorFlowLite/SDK/Scripts/ |
| D | Interpreter.cs | 231 int input_index); in TfLiteInterpreterGetInputTensor()
236 int input_index, in TfLiteInterpreterResizeInputTensor()
|