Home
last modified time | relevance | path

Searched defs:input_index (Results 1 – 25 of 84) sorted by relevance

1234

/external/libtextclassifier/native/utils/
Dtflite-model-executor.h79 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()
Dtflite-model-executor.cc346 void TfLiteModelExecutor::SetInput(const int input_index, in SetInput()
/external/tensorflow/tensorflow/lite/testing/nnapi_tflite_zip_tests/
Dparse_testdata.cc137 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/compiler/tf2xla/
Dxla_helpers.h112 int input_index; member
121 int input_index; member
Dtf2xla.cc118 int64 input_index = xla_args.size() - config.variable_size(); in ConvertGraphToXla() local
/external/tensorflow/tensorflow/lite/testing/
Dparse_testdata.cc134 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/lite/toco/graph_transformations/
Dpropagate_fake_quant_num_bits.cc124 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
Dremove_trivial_passthrough.cc72 int input_index) { in RemoveTrivialPassthroughOp()
Dquantize.cc226 std::size_t input_index, ArrayDataType* quantized_data_type, in ChooseQuantizationForOperatorInput()
573 for (std::size_t input_index = 0; input_index < op.inputs.size(); in Run() local
/external/XNNPACK/test/
Ddepth-to-space-operator-tester.h175 const size_t input_index = in TestNHWCxX32() local
241 const size_t input_index = in TestNCHW2NHWCxX32() local
Ddepthtospace-microkernel-tester.h129 const size_t input_index = in Test() local
/external/tensorflow/tensorflow/core/kernels/
Ddynamic_stitch_op_gpu.cu.cc42 const int32 input_index = data_indices[slice_id]; in DynamicStitchKernel() local
Dquantized_batch_norm_op.cc55 const int input_index = (row_index * depth) + channel; in ReferenceBatchNorm() local
148 const int input_index = (row_index * depth) + channel; in FixedPointBatchNorm() local
Dops_testutil.cc191 TensorValue OpsTestBase::mutable_input(int input_index) { in mutable_input()
/external/tensorflow/tensorflow/compiler/xla/service/
Doptimize_input_output_buffer_alias.cc74 const ShapeIndex& input_index = entry.index; in Build() local
/external/tensorflow/tensorflow/lite/kernels/internal/reference/
Dspace_to_depth.h64 const int input_index = Offset(input_shape, in_b, in_h, in_w, in_d); in SpaceToDepth() local
Ddepth_to_space.h65 const int input_index = Offset(input_shape, in_b, in_h, in_w, in_d); in DepthToSpace() local
Dprelu.h46 int input_index = SubscriptToIndex(desc1, b, y, x, c); in BroadcastPrelu4DSlow() local
/external/tensorflow/tensorflow/lite/c/
Dc_api.cc150 const TfLiteInterpreter* interpreter, int32_t input_index) { in TfLiteInterpreterGetInputTensor()
155 int32_t input_index, in TfLiteInterpreterResizeInputTensor()
/external/tensorflow/tensorflow/compiler/tf2tensorrt/
Dtensorrt_test.cc105 const int input_index = engine.getBindingIndex(kInputTensor); in Execute() local
/external/tensorflow/tensorflow/lite/delegates/gpu/common/
Dquantization_util.cc33 TfLiteContext* context, int input_index, in DequantizeInput()
/external/tensorflow/tensorflow/c/eager/parallel_device/
Dparallel_device_lib.cc238 for (int input_index = 0; input_index < inputs.size(); ++input_index) { in Execute() local
329 for (int input_index = 0; input_index < inputs.size(); ++input_index) { in StartExecute() local
/external/tensorflow/tensorflow/core/kernels/data/experimental/
Dcompute_batch_size_op.cc75 int64 input_index) { in GetInputNode()
/external/tensorflow/tensorflow/lite/experimental/examples/unity/TensorFlowLitePlugin/Assets/TensorFlowLite/SDK/Scripts/
DInterpreter.cs231 int input_index); in TfLiteInterpreterGetInputTensor()
236 int input_index, in TfLiteInterpreterResizeInputTensor()
/external/tensorflow/tensorflow/core/kernels/image/
Dscale_and_translate_op.cc138 int input_index = starts_vec(output_index); in ComputeGradSpansCore() local
170 for (int input_index = 0; input_index < forward_input_size; ++input_index) { in ComputeGradSpansCore() local
251 void GetValues(OpKernelContext* context, int input_index, float* v_1, in GetValues()

1234