Home
last modified time | relevance | path

Searched refs:packed_tensor (Results 1 – 6 of 6) sorted by relevance

/external/ComputeLibrary/src/gpu/cl/utils/
DClAuxTensorHandler.h51 …ICLTensor *packed_tensor = utils::cast::polymorphic_downcast<ICLTensor *>(pack.get_tensor(slot_id)… in _tensor() local
52 if((packed_tensor == nullptr) || (info.total_size() > packed_tensor->info()->total_size())) in _tensor()
69 _tensor.allocator()->import_memory(packed_tensor->cl_buffer()); in _tensor()
/external/ComputeLibrary/src/cpu/utils/
DCpuAuxTensorHandler.h51 … ITensor *packed_tensor = utils::cast::polymorphic_downcast<ITensor *>(pack.get_tensor(slot_id)); in _tensor() local
52 if((packed_tensor == nullptr) || (info.total_size() > packed_tensor->info()->total_size())) in _tensor()
69 _tensor.allocator()->import_memory(packed_tensor->buffer()); in _tensor()
/external/tensorflow/tensorflow/python/
Dpywrap_dtensor_device.cc232 TFE_TensorHandle* packed_tensor; in PYBIND11_MODULE() local
235 packed_tensor = SparsePack( in PYBIND11_MODULE()
252 packed_tensor = Pack(ctx, input_vector.size(), input_vector.data(), in PYBIND11_MODULE()
259 PyList_SET_ITEM(flat_result.get(), 0, EagerTensorFromHandle(packed_tensor)); in PYBIND11_MODULE()
Dtfe_wrapper.cc327 PyObject* packed_tensor = in InputTFE_OutputTensorHandles() local
329 return tensorflow::PyoOrThrow(packed_tensor); in InputTFE_OutputTensorHandles()
/external/tensorflow/tensorflow/dtensor/cc/
Ddtensor_device.cc832 std::unique_ptr<TensorWithLayout> packed_tensor; in Pack() local
844 packed_tensor = TensorWithLayout::Dummy( in Pack()
902 packed_tensor = in Pack()
908 RecordInShapeLayoutCache(*packed_tensor); in Pack()
910 MakeLayoutTensorHandle(context, std::move(packed_tensor), status); in Pack()
990 std::unique_ptr<TensorWithLayout> packed_tensor; in SparsePack() local
993 packed_tensor = SparseTensorWithLayout::Dummy( in SparsePack()
1045 packed_tensor = in SparsePack()
1054 RecordInShapeLayoutCache(*packed_tensor); in SparsePack()
1056 MakeLayoutTensorHandle(context, std::move(packed_tensor), status); in SparsePack()
/external/tensorflow/tensorflow/python/framework/
Dops.py1554 packed_tensor = ctx.pack_eager_tensors(tensors)
1556 packed_tensor._handle_data = handle_data # pylint: disable=protected-access
1562 tape.record_operation("pack_eager_tensors", [packed_tensor], tensors,
1565 return packed_tensor