Home
last modified time | relevance | path

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

1234

/external/libtextclassifier/utils/
Dtflite-model-executor.h68 void SetInput(const int input_index, const TensorView<T>& input_data, in SetInput()
75 void SetInput(const int input_index, const std::vector<T>& input_data, in SetInput()
82 void SetInput(const int input_index, const T input_value, in SetInput()
Dtflite-model-executor.cc213 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/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.cc128 bool DoesOpInputBlockBackwardPropagation(const Operator& op, int input_index) { in DoesOpInputBlockBackwardPropagation()
156 for (int input_index = 0; input_index < op->inputs.size(); ++input_index) { in RecursivelyBackwardPropagateDataType() local
Dremove_trivial_passthrough.cc66 int input_index) { in RemoveTrivialPassthroughOp()
Dquantize.cc185 std::size_t input_index, ArrayDataType* quantized_data_type, in ChooseQuantizationForOperatorInput()
524 for (std::size_t input_index = 0; input_index < op.inputs.size(); in Run() local
/external/tensorflow/tensorflow/core/kernels/
Dops_testutil.h188 const Tensor& GetInput(int input_index) const { in GetInput()
194 TensorValue mutable_input(int input_index) { in mutable_input()
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
/external/tensorflow/tensorflow/lite/experimental/examples/unity/TensorFlowLitePlugin/Assets/TensorFlowLite/SDK/Scripts/
DInterpreter.cs119 int input_index); in TFL_InterpreterGetInputTensor()
124 int input_index, in TFL_InterpreterResizeInputTensor()
/external/tensorflow/tensorflow/compiler/xla/service/
Doptimize_input_output_buffer_alias.cc72 const ShapeIndex& input_index = it->second.front(); in Build() local
/external/tensorflow/tensorflow/contrib/tensor_forest/kernels/
Dscatter_add_ndim_op.cc99 const int32 input_index = start_index + offset; in Compute() local
Dtree_utils.h212 int32 GetNumSparseFeatures(const T1& indices, int32 input_index, in GetNumSparseFeatures()
/external/v8/src/compiler/
Dnode.h185 int input_index() const { return InputIndexField::decode(bit_field_); } in input_index() function
241 Node* const* GetInputPtrConst(int input_index) const { in GetInputPtrConst()
245 Node** GetInputPtr(int input_index) { in GetInputPtr()
249 Use* GetUsePtr(int input_index) { in GetUsePtr()
Dint64-lowering.h67 int input_index; member
Dsimd-scalar-lowering.h52 int input_index; member
Dgraph-reducer.h148 int input_index; in NON_EXPORTED_BASE() member
/external/tensorflow/tensorflow/lite/experimental/c/
Dc_api.cc135 int32_t input_index) { in TFL_InterpreterGetInputTensor()
140 int32_t input_index, in TFL_InterpreterResizeInputTensor()
/external/tensorflow/tensorflow/compiler/xrt/
Dxrt.proto81 int32 input_index = 1; field
/external/tensorflow/tensorflow/python/ops/
Dspecial_math_ops.py343 def sort_key(input_index, a): argument
/external/tensorflow/tensorflow/core/grappler/utils/
Dfunctions.h128 int input_index; // Index of a tensor in the function input argument member
/external/tensorflow/tensorflow/compiler/tf2tensorrt/
Dtensorrt_test.cc106 const int input_index = engine.getBindingIndex(kInputTensor); in Execute() local
/external/tensorflow/tensorflow/lite/python/
Dinterpreter.py189 def resize_tensor_input(self, input_index, tensor_size): argument
/external/tensorflow/tensorflow/compiler/tf2xla/
Dxla_compiler.h225 int input_index; member
231 int input_index; member

1234