Home
last modified time | relevance | path

Searched refs:wrapped_tensor (Results 1 – 9 of 9) sorted by relevance

/external/tensorflow/tensorflow/lite/kernels/shim/
Dtflite_tensor_view.h48 TfLiteTensorView(::TfLiteTensor *wrapped_tensor, const DType &dtype) in TfLiteTensorView() argument
49 : TensorView(absl::Span<int>(wrapped_tensor->dims->data, in TfLiteTensorView()
50 wrapped_tensor->dims->size), in TfLiteTensorView()
51 wrapped_tensor->data.raw, wrapped_tensor->bytes, dtype), in TfLiteTensorView()
52 wrapped_tensor_(wrapped_tensor), in TfLiteTensorView()
53 const_wrapped_tensor_(wrapped_tensor) {} in TfLiteTensorView()
56 TfLiteTensorView(::TfLiteTensor *wrapped_tensor,
61 TfLiteTensorView(const ::TfLiteTensor *wrapped_tensor, const DType &dtype) in TfLiteTensorView() argument
62 : TensorView(absl::Span<int>(wrapped_tensor->dims->data, in TfLiteTensorView()
63 wrapped_tensor->dims->size), in TfLiteTensorView()
[all …]
Dtflite_tensor_view.cc31 return TfLiteTensorView(wrapped_tensor, DType()); \
41 TfLiteTensorView::TfLiteTensorView(::TfLiteTensor *wrapped_tensor, in TfLiteTensorView() argument
43 : TensorView(absl::Span<int>(wrapped_tensor->dims->data, in TfLiteTensorView()
44 wrapped_tensor->dims->size), in TfLiteTensorView()
46 wrapped_tensor_(wrapped_tensor), in TfLiteTensorView()
47 const_wrapped_tensor_(wrapped_tensor) { in TfLiteTensorView()
51 TfLiteTensorView::TfLiteTensorView(const ::TfLiteTensor *wrapped_tensor, in TfLiteTensorView() argument
53 : TensorView(absl::Span<int>(wrapped_tensor->dims->data, in TfLiteTensorView()
54 wrapped_tensor->dims->size), in TfLiteTensorView()
56 const_wrapped_tensor_(wrapped_tensor) { in TfLiteTensorView()
[all …]
Dtf_tensor_view.h46 TfTensorView(const ::tensorflow::Tensor *wrapped_tensor, const DType &dtype);
52 TfTensorViewTemplatizedNew(TfTensorType *wrapped_tensor);
73 ::tensorflow::Tensor *wrapped_tensor);
78 const ::tensorflow::Tensor *wrapped_tensor);
85 TfTensorView::TfTensorView(const ::tensorflow::Tensor *wrapped_tensor,
87 : TensorView({}, wrapped_tensor->data(),
88 wrapped_tensor->tensor_data().size(), dtype) {
89 shape_data_.resize(wrapped_tensor->shape().dims());
90 for (int dim = 0; dim < wrapped_tensor->shape().dims(); ++dim) {
91 shape_data_[dim] = wrapped_tensor->shape().dim_size(dim);
Dtf_tensor_view.cc27 return TfTensorView(wrapped_tensor, DType()); \
66 TfTensorViewTemplatizedNew(TfTensorType *wrapped_tensor) { in TfTensorViewTemplatizedNew() argument
67 switch (wrapped_tensor->dtype()) { in TfTensorViewTemplatizedNew()
81 absl::StrCat("Unsupported data type: ", wrapped_tensor->dtype())); in TfTensorViewTemplatizedNew()
88 ::tensorflow::Tensor *wrapped_tensor) { in New() argument
89 return TfTensorViewTemplatizedNew(wrapped_tensor); in New()
94 const ::tensorflow::Tensor *wrapped_tensor) { in New()
95 return TfTensorViewTemplatizedNew(wrapped_tensor); in New()
Dtensor_view.h149 W *wrapped_tensor);
/external/tensorflow/tensorflow/python/ops/ragged/
Dragged_tensor_supported_values_test.py424 wrapped_tensor = WrappedTensor(constant_op.constant(flat_values))
425 rt = RaggedTensor.from_nested_row_splits(wrapped_tensor, nested_row_splits)
428 self.assertAllEqual(components[0].value, wrapped_tensor.value)
433 wrapped_tensor.value)
/external/pytorch/aten/src/ATen/native/sparse/
DSparseTensor.cpp741 const auto wrapped_tensor = [](const Tensor& t, in sparse_mask_like_prepare_sparse_inputs() local
767 auto res = wrapped_tensor(t, res_indices, res_values)._coalesced_(true); in sparse_mask_like_prepare_sparse_inputs()
772 const auto rhs = mask.is_coalesced() ? wrapped_tensor(mask) : mask; in sparse_mask_like_prepare_sparse_inputs()
/external/pytorch/test/
Dtest_serialization.py4201 wrapped_tensor = torch.rand(2)
4202 my_tensor = TestWrapperSubclass(wrapped_tensor)
4218 wrapped_tensor = torch.rand(2)
4219 my_tensor = TestGetStateSubclass(wrapped_tensor)
4236 wrapped_tensor = torch.rand(2)
4237 my_tensor = TestWrapperSubclass(wrapped_tensor)
/external/tensorflow/tensorflow/python/ops/parallel_for/
Dpfor.py1397 def _unwrap_or_tile(self, wrapped_tensor): argument
1399 output, is_stacked = wrapped_tensor.t, wrapped_tensor.is_stacked