Home
last modified time | relevance | path

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

12345

/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()
/external/XNNPACK/test/
Ddepth-to-space-operator-tester.h177 const size_t input_index = in TestNHWCxX8() local
243 const size_t input_index = in TestNHWCxX16() local
309 const size_t input_index = in TestNHWCxX32() local
375 const size_t input_index = in TestNCHW2NHWCxX32() local
Dspace-to-depth-operator-tester.h173 const size_t input_index = oc in TestNHWCxX8() local
242 const size_t input_index = oc in TestNHWCxX16() local
311 const size_t input_index = oc in TestNHWCxX32() local
Dconstant-pad-operator-tester.h163 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
/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.cc139 for (int input_index = 0; input_index < inputs; input_index++) { in ParseExamples() local
147 for (int input_index = 0; input_index < outputs; input_index++) { in ParseExamples() local
160 int input_index = interpreter->inputs()[i]; in FeedExample() local
168 int input_index = interpreter->inputs()[i]; in FeedExample() local
/external/tensorflow/tensorflow/compiler/tf2xla/
Dxla_helpers.h106 int input_index; member
115 int input_index; member
Dtf2xla.cc116 int64_t input_index = xla_args.size() - config.variable_size(); in ConvertGraphToXla() 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()
576 for (std::size_t input_index = 0; input_index < op.inputs.size(); in Run() local
/external/tensorflow/tensorflow/compiler/mlir/lite/transforms/
Dprepare_quantize_helper.h171 LogicalResult processRestrictScale(SourceOp op, int input_index, in processRestrictScale()
280 SourceOp op, Operation* const_op, int input_index, in processConstantOp()
335 SourceOp op, quantfork::StatisticsOp stats_op, int input_index, in replaceStatsOp()
/external/rust/crates/base64/src/engine/
Dnaive.rs55 let mut input_index = 0_usize; in internal_encode() localVariable
130 let mut input_index = 0_usize; in internal_decode() localVariable
/external/tensorflow/tensorflow/core/kernels/
Ddynamic_stitch_op_gpu.cu.cc42 const int32 input_index = data_indices[slice_id]; in DynamicStitchKernel() local
Dreshape_util_gpu.cu.cc37 const Tindex* input_index = &input_indices[sparse_index * input_rank]; in ReshapeSparseTensorKernel() 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/c/
Dc_api_experimental.cc109 const TfLiteInterpreter* interpreter, int32_t input_index) { in TfLiteInterpreterGetInputTensorIndex()
146 const TfLiteSignatureRunner* signature_runner, const int32_t input_index) { in TfLiteSignatureRunnerGetInputName()
/external/tensorflow/tensorflow/python/framework/
Dnode_file_writer_test.py123 def _get_input_tensor(self, node_def, input_index): argument
/external/tensorflow/tensorflow/lite/kernels/internal/reference/
Ddepth_to_space.h65 const int input_index = Offset(input_shape, in_b, in_h, in_w, in_d); in DepthToSpace() local
Dspace_to_depth.h66 const int input_index = Offset(input_shape, in_b, in_h, in_w, in_d); in SpaceToDepth() local
Dprelu.h48 int input_index = SubscriptToIndex(desc1, b, y, x, c); in BroadcastPrelu4DSlow() local
/external/tensorflow/tensorflow/compiler/xla/service/
Doptimize_input_output_buffer_alias.cc74 const ShapeIndex& input_index = entry.index; in Build() local
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dmark_input_output_aliases.cc38 int input_index; member
/external/tensorflow/tensorflow/lite/delegates/gpu/common/
Dquantization_util.cc33 TfLiteContext* context, int input_index, in DequantizeInput()
/external/tensorflow/tensorflow/lite/experimental/examples/unity/TensorFlowLitePlugin/Assets/TensorFlowLite/SDK/Scripts/
DInterpreter.cs233 int input_index); in TfLiteInterpreterGetInputTensor()
238 int input_index, in TfLiteInterpreterResizeInputTensor()

12345