Home
last modified time | relevance | path

Searched defs:weights_tensor (Results 1 – 16 of 16) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/lite/tools/optimize/
H A Dquantize_weights_test.cc380 const auto weights_tensor = in TEST_F() local
417 const auto weights_tensor = in TEST_F() local
455 const auto weights_tensor = in TEST_F() local
496 const auto weights_tensor = in TEST_F() local
543 const auto weights_tensor = in TEST_F() local
H A Dquantization_utils_test.cc549 TensorT* weights_tensor = subgraph->tensors.at(weights_tensor_idx).get(); in TEST_F() local
599 TensorT* weights_tensor = subgraph->tensors.at(weights_tensor_idx).get(); in TEST_F() local
634 TensorT* weights_tensor = subgraph->tensors.at(weights_tensor_idx).get(); in TEST_F() local
702 TensorT* weights_tensor = subgraph->tensors.at(weights_tensor_idx).get(); in TEST_F() local
H A Dquantize_model_test.cc633 const auto weights_tensor = in TEST_F() local
744 const auto weights_tensor = in TEST_P() local
848 const auto weights_tensor = in TEST_P() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/lite/quantization/lite/
H A Dquantize_weights_test.cc451 const auto weights_tensor = in TEST_F() local
487 const auto weights_tensor = in TEST_F() local
524 const auto weights_tensor = in TEST_F() local
564 const auto weights_tensor = in TEST_F() local
610 const auto weights_tensor = in TEST_F() local
H A Dquantize_model_test.cc437 const auto weights_tensor = in TEST_P() local
542 const auto weights_tensor = in TEST_P() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/hexagon/builders/
H A Dmatmul_builder.cc159 const auto& weights_tensor = context->tensors[weights_tensor_id]; in PopulateSubGraph() local
227 const auto& weights_tensor = context->tensors[weights_tensor_id]; in PopulateSubGraph() local
H A Dconv_2d_helpers.cc47 const TfLiteTensor& weights_tensor, TfLiteContext* context, in ProcessPerChannelQuantizedWeights()
229 const auto& weights_tensor = context->tensors[inputs->data[1]]; in InitializeWeightsNodes() local
H A Dtranspose_conv_2d_builder.cc49 const auto& weights_tensor = context->tensors[tensor_id]; in PopulateSubGraph() local
H A Dop_builder.cc99 const auto& weights_tensor = context_->tensors[node->inputs->data[1]]; in CreateOpBuilderFromTfLiteOp() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/common/selectors/
H A Doperation_selector.cc279 TensorDescriptor weights_tensor = TensorDescriptor( in AddConvSharedWeights() local
291 TensorDescriptor weights_tensor = TensorDescriptor( in AddConvSharedWeights() local
/aosp_15_r20/external/ComputeLibrary/src/runtime/
H A DIWeightsManager.cpp63 ITensor *weights_tensor{ nullptr }; in run() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/hexagon/builders/tests/
H A Dmatmul_test.cc61 auto* weights_tensor = interpreter_->tensor(weights_); in FullyConnectedOpModel() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/common/
H A Dlstm_parser.cc57 const TfLiteTensor* weights_tensor = in GetFullyConnectedNode() local
H A Dmodel_builder.cc593 const TfLiteTensor* weights_tensor = reader->GetInputTensor(1); in Parse() local
645 const TfLiteTensor* weights_tensor = reader->GetInputTensor(1); in ReadAttributes() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/lite/ir/
H A Dtfl_ops.cc1079 DenseElementsAttr input_tensor, weights_tensor, bias_tensor; in fold() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/nnapi/
H A Dnnapi_delegate.cc271 const auto& weights_tensor = context->tensors[weights_id]; in NeedInt8Conversion() local