Home
last modified time | relevance | path

Searched defs:wrapped_tensor (Results 1 – 5 of 5) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/shim/
H A Dtflite_tensor_view.cc41 TfLiteTensorView::TfLiteTensorView(::TfLiteTensor *wrapped_tensor, in TfLiteTensorView()
51 TfLiteTensorView::TfLiteTensorView(const ::TfLiteTensor *wrapped_tensor, in TfLiteTensorView()
124 TfLiteTensorViewTemplatizedNew(TfLiteTensorType *wrapped_tensor) { in TfLiteTensorViewTemplatizedNew()
147 ::TfLiteTensor *wrapped_tensor) { in New()
152 absl::StatusOr<const TfLiteTensorView> TensorView::New<const ::TfLiteTensor>( in New()
H A Dtf_tensor_view.cc66 TfTensorViewTemplatizedNew(TfTensorType *wrapped_tensor) { in TfTensorViewTemplatizedNew()
88 ::tensorflow::Tensor *wrapped_tensor) { in New()
93 absl::StatusOr<const TfTensorView> TensorView::New<const ::tensorflow::Tensor>( in New()
H A Dtflite_tensor_view.h48 TfLiteTensorView(::TfLiteTensor *wrapped_tensor, const DType &dtype) in TfLiteTensorView()
61 TfLiteTensorView(const ::TfLiteTensor *wrapped_tensor, const DType &dtype) in TfLiteTensorView()
/aosp_15_r20/external/pytorch/aten/src/ATen/native/sparse/
H A DSparseTensor.cpp741 const auto wrapped_tensor = [](const Tensor& t, in sparse_mask_like_prepare_sparse_inputs() local
/aosp_15_r20/external/tensorflow/tensorflow/python/ops/parallel_for/
H A Dpfor.py1397 def _unwrap_or_tile(self, wrapped_tensor): argument