Home
last modified time | relevance | path

Searched refs:strong_shape_tensors_ (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/lite/delegates/gpu/cl/
Dinference_context.cc605 const auto& it = strong_shape_tensors_.find(id); in AllocateMemoryForStrongShapes()
606 if (it == strong_shape_tensors_.end()) { in AllocateMemoryForStrongShapes()
608 &strong_shape_tensors_[id])); in AllocateMemoryForStrongShapes()
687 for (const auto& t : strong_shape_tensors_) { in GetSizeOfMemoryAllocatedForIntermediateTensors()
709 return &strong_shape_tensors_[graph_ids_to_strong_shape_tensors_[id]]; in GetTensor()
Dinference_context.h229 std::map<ValueId, Tensor> strong_shape_tensors_; variable
/external/tensorflow/tensorflow/lite/delegates/gpu/metal/
Dinference_context.cc395 return &strong_shape_tensors_ in GetTensor()
576 const auto& it = strong_shape_tensors_.find(id); in AllocateMemoryForStrongShapes()
577 if (it == strong_shape_tensors_.end()) { in AllocateMemoryForStrongShapes()
580 &strong_shape_tensors_[id])); in AllocateMemoryForStrongShapes()
Dinference_context.h222 std::map<ValueId, MetalSpatialTensor> strong_shape_tensors_; variable