/aosp_15_r20/external/tensorflow/tensorflow/lite/tools/optimize/ |
H A D | quantize_weights_test.cc | 380 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 D | quantization_utils_test.cc | 549 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 D | quantize_model_test.cc | 633 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 D | quantize_weights_test.cc | 451 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 D | quantize_model_test.cc | 437 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 D | matmul_builder.cc | 159 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 D | conv_2d_helpers.cc | 47 const TfLiteTensor& weights_tensor, TfLiteContext* context, in ProcessPerChannelQuantizedWeights() 229 const auto& weights_tensor = context->tensors[inputs->data[1]]; in InitializeWeightsNodes() local
|
H A D | transpose_conv_2d_builder.cc | 49 const auto& weights_tensor = context->tensors[tensor_id]; in PopulateSubGraph() local
|
H A D | op_builder.cc | 99 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 D | operation_selector.cc | 279 TensorDescriptor weights_tensor = TensorDescriptor( in AddConvSharedWeights() local 291 TensorDescriptor weights_tensor = TensorDescriptor( in AddConvSharedWeights() local
|
/aosp_15_r20/external/ComputeLibrary/src/runtime/ |
H A D | IWeightsManager.cpp | 63 ITensor *weights_tensor{ nullptr }; in run() local
|
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/hexagon/builders/tests/ |
H A D | matmul_test.cc | 61 auto* weights_tensor = interpreter_->tensor(weights_); in FullyConnectedOpModel() local
|
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/common/ |
H A D | lstm_parser.cc | 57 const TfLiteTensor* weights_tensor = in GetFullyConnectedNode() local
|
H A D | model_builder.cc | 593 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 D | tfl_ops.cc | 1079 DenseElementsAttr input_tensor, weights_tensor, bias_tensor; in fold() local
|
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/nnapi/ |
H A D | nnapi_delegate.cc | 271 const auto& weights_tensor = context->tensors[weights_id]; in NeedInt8Conversion() local
|