Home
last modified time | relevance | path

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

1234

/external/libtextclassifier/native/utils/
Dtflite-model-executor.h70 void SetInput(const int input_index, const TensorView<T>& input_data, in SetInput()
77 void SetInput(const int input_index, const std::vector<T>& input_data, in SetInput()
84 void SetInput(const int input_index, const T input_value, in SetInput()
Dtflite-model-executor.cc248 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.cc124 bool DoesOpInputBlockBackwardPropagation(const Operator& op, int input_index) { in DoesOpInputBlockBackwardPropagation()
152 for (int 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()
566 for (std::size_t input_index = 0; input_index < op.inputs.size(); in Run() local
/external/tensorflow/tensorflow/lite/experimental/examples/unity/TensorFlowLitePlugin/Assets/TensorFlowLite/SDK/Scripts/
DInterpreter.cs119 int input_index); in TfLiteInterpreterGetInputTensor()
124 int input_index, in TfLiteInterpreterResizeInputTensor()
/external/tensorflow/tensorflow/lite/kernels/internal/reference/
Dprelu.h48 int input_index = SubscriptToIndex(desc1, b, y, x, c); in BroadcastPrelu4DSlow() local
/external/tensorflow/tensorflow/compiler/tf2xla/
Dtf2xla.cc115 int64 input_index = xla_args.size() - config.variable_size(); in ConvertGraphToXla() local
Dxla_compiler.h232 int input_index; member
241 int input_index; member
/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.cc180 TensorValue OpsTestBase::mutable_input(int input_index) { in mutable_input()
Ddeep_conv2d_test.cc30 const int input_index = k * cols + l; in ComputeKroneckerProduct() local
Dscale_and_translate_op.cc137 int input_index = starts_vec(output_index); in ComputeGradSpansCore() local
169 for (int input_index = 0; input_index < forward_input_size; ++input_index) { in ComputeGradSpansCore() local
250 void GetValues(OpKernelContext* context, int input_index, float* v_1, in GetValues()
/external/v8/src/compiler/
Dnode.h186 int input_index() const { return InputIndexField::decode(bit_field_); } in input_index() function
254 ZoneNodePtr const* GetInputPtrConst(int input_index) const { in GetInputPtrConst()
258 ZoneNodePtr* GetInputPtr(int input_index) { in GetInputPtr()
262 Use* GetUsePtr(int input_index) { in GetUsePtr()
Dint64-lowering.h82 int input_index; member
Dsimd-scalar-lowering.h60 int input_index; member
Dgraph-reducer.h158 int input_index; in NON_EXPORTED_BASE() member
/external/tensorflow/tensorflow/lite/c/
Dc_api.cc152 const TfLiteInterpreter* interpreter, int32_t input_index) { in TfLiteInterpreterGetInputTensor()
157 int32_t input_index, in TfLiteInterpreterResizeInputTensor()
/external/tensorflow/tensorflow/compiler/xla/service/
Doptimize_input_output_buffer_alias.cc79 const ShapeIndex& input_index = entry.index; in Build() local
/external/tensorflow/tensorflow/compiler/tf2tensorrt/
Dtensorrt_test.cc106 const int input_index = engine.getBindingIndex(kInputTensor); in Execute() local
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dpromote_resources_to_args.cc53 int64_t input_index; member
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Delemental_ir_emitter.cc408 IrArray::Index input_index(input_multi_index, operand->shape(), in MakeElementGenerator() local
450 llvm_ir::IrArray::Index input_index( in MakeElementGenerator() local

1234