Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/lite/tools/optimize/
Dquantize_weights_test.cc374 const auto weights_tensor = in TEST_F() local
410 const auto weights_tensor = in TEST_F() local
447 const auto weights_tensor = in TEST_F() local
487 const auto weights_tensor = in TEST_F() local
533 const auto weights_tensor = in TEST_F() local
Dquantization_utils_test.cc479 TensorT* weights_tensor = subgraph->tensors.at(weights_tensor_idx).get(); in TEST_F() local
529 TensorT* weights_tensor = subgraph->tensors.at(weights_tensor_idx).get(); in TEST_F() local
564 TensorT* weights_tensor = subgraph->tensors.at(weights_tensor_idx).get(); in TEST_F() local
633 TensorT* weights_tensor = subgraph->tensors.at(weights_tensor_idx).get(); in TEST_F() local
Dquantize_model_test.cc618 const auto weights_tensor = in TEST_F() local
727 const auto weights_tensor = in TEST_P() local
/external/tensorflow/tensorflow/lite/delegates/hexagon/builders/
Dconv_2d_helpers.cc46 const auto& weights_tensor = context->tensors[inputs->data[1]]; in ProcessPerChannelQuantizedWeights() local
143 const auto& weights_tensor = context->tensors[inputs->data[1]]; in InitializeWeightsNodes() local
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
Dtranspose_conv_2d_builder.cc49 const auto& weights_tensor = context->tensors[tensor_id]; in PopulateSubGraph() local
Dop_builder.cc99 const auto& weights_tensor = context_->tensors[node->inputs->data[1]]; in CreateOpBuilderFromTfLiteOp() local
/external/tensorflow/tensorflow/lite/delegates/hexagon/builders/tests/
Dmatmul_test.cc62 auto* weights_tensor = interpreter_->tensor(weights_); in FullyConnectedOpModel() local
/external/tensorflow/tensorflow/compiler/mlir/lite/ir/
Dtfl_ops.cc850 DenseElementsAttr input_tensor, weights_tensor, bias_tensor; in fold() local
/external/tensorflow/tensorflow/lite/delegates/nnapi/
Dnnapi_delegate.cc245 const auto& weights_tensor = context->tensors[weights_id]; in NeedInt8Conversion() local