Home
last modified time | relevance | path

Searched defs:input_index (Results 1 – 25 of 115) 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/mesa3d/src/gallium/targets/teflon/
Dtest_executor.cpp59 unsigned input_index; in patch_conv2d() local
181 auto input_index = add_op->inputs.data()[0]; in patch_quant_for_add() local
211 unsigned input_index = subgraph->operators[2]->inputs.data()[0]; in add_generate_model() 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/mesa3d/src/gallium/drivers/radeonsi/
Dsi_nir_lower_vs_inputs.c40 get_vertex_index(nir_builder *b, int input_index, struct lower_vs_inputs_state *s) in get_vertex_index()
100 load_vs_input_from_blit_sgpr(nir_builder *b, unsigned input_index, in load_vs_input_from_blit_sgpr()
428 load_vs_input_from_vertex_buffer(nir_builder *b, unsigned input_index, in load_vs_input_from_vertex_buffer()
569 unsigned input_index = nir_intrinsic_base(intrin); in lower_vs_input_instr() local
/external/rust/crates/base64/src/engine/
Dnaive.rs54 let mut input_index = 0_usize; in internal_encode() localVariable
128 let input_index = chunk_index * 4; 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
/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/lite/kernels/internal/reference/
Dspace_to_depth.h66 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.h48 int input_index = SubscriptToIndex(desc1, b, y, x, c); in BroadcastPrelu4DSlow() local
/external/tensorflow/tensorflow/python/framework/
Dnode_file_writer_test.py123 def _get_input_tensor(self, node_def, input_index): argument
/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()

12345