Home
last modified time | relevance | path

Searched refs:weights_tensor_id (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/lite/delegates/hexagon/builders/
Dmatmul_builder.cc158 int weights_tensor_id = inputs->data[1]; in PopulateSubGraph() local
159 const auto& weights_tensor = context->tensors[weights_tensor_id]; in PopulateSubGraph()
198 graph_builder_->AddTensorWithID(weights_tensor_id, in PopulateSubGraph()
209 inputs, outputs, graph_builder_->GetHexagonTensorId(weights_tensor_id), in PopulateSubGraph()
226 const int weights_tensor_id = inputs->data[1]; in PopulateSubGraph() local
227 const auto& weights_tensor = context->tensors[weights_tensor_id]; in PopulateSubGraph()
238 AddInput(graph_builder_->GetHexagonTensorId(weights_tensor_id)); in PopulateSubGraph()
/external/tensorflow/tensorflow/lite/delegates/gpu/common/
Dlstm_parser.cc55 absl::Status SetFullyConnectedWeights(int weights_tensor_id, in SetFullyConnectedWeights() argument
59 RETURN_IF_ERROR(reader->ReadTensor(weights_tensor_id, &weights)); in SetFullyConnectedWeights()
Dmodel_builder.cc59 absl::Status GetFullyConnectedAttributes(int weights_tensor_id, in GetFullyConnectedAttributes() argument
64 RETURN_IF_ERROR(reader->ReadTensor(weights_tensor_id, &weights)); in GetFullyConnectedAttributes()