Home
last modified time | relevance | path

Searched defs:tensor_idx (Results 1 – 25 of 29) sorted by relevance

12

/aosp_15_r20/external/tensorflow/tensorflow/lite/tools/optimize/
H A Dquantize_weights.cc74 int32_t tensor_idx) { in GetTensorConsumers()
192 int32_t tensor_idx = op->inputs[op_input_idx]; in CheckAllOpInputsQuantized() local
222 int32_t tensor_idx = op->inputs[op_input_idx]; in InsertQuantizableInputTensorsFromOperator() local
429 int32_t tensor_idx = tensor_pair.first; local
535 for (auto tensor_idx : op->inputs) { in QuantizeWeightsFloat16() local
559 int32_t tensor_idx = tensor_pair.first; in QuantizeWeightsFloat16() local
H A Dquantize_weights_portable.cc60 int32_t tensor_idx) { in GetTensorConsumers()
178 int32_t tensor_idx = op->inputs[op_input_idx]; in CheckAllOpInputsQuantized() local
208 int32_t tensor_idx = op->inputs[op_input_idx]; in InsertQuantizableInputTensorsFromOperator() local
415 int32_t tensor_idx = tensor_pair.first; local
521 for (auto tensor_idx : op->inputs) { in QuantizeWeightsFloat16() local
545 int32_t tensor_idx = tensor_pair.first; in QuantizeWeightsFloat16() local
H A Dquantize_model.cc134 bool IsFloatTensor(const SubGraphT* subgraph, int32_t tensor_idx) { in IsFloatTensor()
241 const int32_t tensor_idx = op->inputs[input_idx]; in PopulateRealValueOpSet() local
250 const int32_t tensor_idx = op->outputs[output_idx]; in PopulateRealValueOpSet() local
258 const int32_t tensor_idx = op->inputs[0]; in PopulateRealValueOpSet() local
265 const int32_t tensor_idx = op->outputs[0]; in PopulateRealValueOpSet() local
411 const int32_t tensor_idx, const TensorType& input_type, in SetInputType()
475 const int32_t tensor_idx, const TensorType& output_type, in SetOutputType()
891 const int32_t tensor_idx = op->inputs[input_idx]; in QuantizeOpInput() local
1619 for (int32_t tensor_idx = 0; tensor_idx < subgraph->tensors.size(); in GetAllOperatorOutputs() local
1662 const int32_t tensor_idx = op->inputs[input_idx]; in FillQuantizationParams() local
H A Dmodify_model_interface.cc405 const int32_t tensor_idx = subgraph->inputs[input_idx]; in AddUint8Dequant() local
445 const int32_t tensor_idx = subgraph->outputs[output_idx]; in AddUint8Quant() local
H A Dquantize_weights_test.cc97 bool IsModelInputOrOutput(const Model* model, uint32_t tensor_idx) { in IsModelInputOrOutput()
117 uint32_t tensor_idx, in GetProducerOpCode()
H A Dquantize_model_test.cc314 for (int tensor_idx = 0; tensor_idx < subgraph->tensors.size(); in TEST_P() local
392 for (int tensor_idx = 0; tensor_idx < subgraph->tensors.size(); in TEST_P() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/tools/optimize/calibration/
H A Dcalibrator_test.cc107 for (int tensor_idx = 0; tensor_idx < 4; tensor_idx++) { in TEST() local
167 for (int tensor_idx = 0; tensor_idx < 7; tensor_idx++) { in TEST() local
186 for (int tensor_idx = 1; tensor_idx < 5; tensor_idx++) { in TEST() local
258 for (int tensor_idx = 0; tensor_idx < 7; tensor_idx++) { in TEST() local
277 for (int tensor_idx = 0; tensor_idx < 7; tensor_idx++) { in TEST() local
/aosp_15_r20/external/executorch/backends/vulkan/test/glsl/
H A Dindexing_utils.h90 ivec4 tensor_idx = pos.xyzz; in to_tensor_idx() local
93 ivec4 tensor_idx = pos.xyzy; in to_tensor_idx() local
96 ivec4 tensor_idx = pos.xyzx; in to_tensor_idx() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/common/
H A Dobject_reader.cc38 uint32_t tensor_idx, Value** value) { in ReadNonConstantTensor()
112 absl::Status ObjectReader::ReadValueByTensorIdx(uint32_t tensor_idx, in ReadValueByTensorIdx()
142 const int tensor_idx = node_->inputs->data[idx]; in GetTensorDims() local
H A Dmodel_builder.cc2916 int tensor_idx = tensor_indices->data[i]; in IsAllAllowedTensors() local
3248 int tensor_idx = tflite_node->inputs->data[i]; in CopyVariableTensorOutputs() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/python/analyzer_wrapper/
H A Dmodel_analyzer.cc69 const std::string tensor_str(const int tensor_idx, const int subgraph_idx, in tensor_str()
102 const tflite::Tensor* tensor, const int tensor_idx, in dump_tensor_detail()
157 const int tensor_idx = tensors->Get(i); in dump_tensor_list() local
/aosp_15_r20/external/tflite-support/tensorflow_lite_support/custom_ops/
H A Dtflite_inference_main.cc62 for (int tensor_idx : interpreter->inputs()) { in RunWithRandomInputs() local
87 for (int tensor_idx : interpreter->outputs()) { in RunWithRandomInputs() local
/aosp_15_r20/external/executorch/backends/vulkan/runtime/graph/ops/glsl/
H A Dindexing_utils.h318 ivec4 tensor_idx = pos.xyzz; in to_tensor_idx() local
321 ivec4 tensor_idx = pos.xyzy; in to_tensor_idx() local
324 ivec4 tensor_idx = pos.xyzx; in to_tensor_idx() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/java/src/main/native/
H A Dnativesignaturerunner_jni.cc65 for (int tensor_idx : signature_runner_->subgraph_->inputs()) { in GetInputIndex() local
78 for (int tensor_idx : signature_runner_->subgraph_->outputs()) { in GetOutputIndex() local
H A Dnativeinterpreterwrapper_jni.cc566 const int tensor_idx = interpreter->inputs()[input_idx]; in Java_org_tensorflow_lite_NativeInterpreterWrapper_resizeInput() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A DNamedTensor.cpp253 const auto tensor_idx = tensor_idx_for[order_idx]; in align_to() local
263 for (const auto tensor_idx : c10::irange(tensor_dim)) { in align_to() local
H A DTensorShape.h60 for (const auto tensor_idx : c10::irange(tensors.size())) { in have_same_ndims() local
H A DCPUFallback.cpp156 auto tensor_idx = tensor_args_indices[i]; in cpu_fallback() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/testing/
H A Dselective_build_test.cc48 for (int tensor_idx : interpreter->inputs()) { in RunWithRandomInputs() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/lite/quantization/lite/
H A Dquantize_weights_test.cc107 bool IsModelInputOrOutput(const Model* model, uint32_t tensor_idx) { in IsModelInputOrOutput()
127 uint32_t tensor_idx, BuiltinOperator* op_code) { in GetProducerOpCode()
/aosp_15_r20/external/pytorch/torch/csrc/lazy/ts_backend/
H A Dts_eager_fallback.cpp261 auto tensor_idx = tensor_args_indices[i]; in ts_eager_fallback() local
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dsparse_conditional_accumulator.h75 const Tensor* tensor_idx = std::get<0>(*tensor); in ValidateShape() local
/aosp_15_r20/external/mesa3d/src/gallium/frontends/teflon/
H A Dtfl_device.c268 unsigned tensor_idx = params->input_tensors->data[i]; in partition_init() local
/aosp_15_r20/external/ComputeLibrary/tests/validation/fixtures/
H A DGEMMFixture.h1179 int tensor_idx = 0; in compute_reference() local
1769 int tensor_idx = 0; in compute_reference() local
2284 int tensor_idx = 0; in compute_reference() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/
H A DTensorShape.cu395 const int64_t tensor_idx = block_idx_to_tensor_idx[blockIdx.x]; in chunk_cat_cuda_kernel() local

12