Searched refs:GetByteSizeRequirement (Results 1 – 9 of 9) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | generic_transfer_manager.cc | 53 stream, GetByteSizeRequirement(shape), element_pointers.data(), region)); in WriteSingleTupleIndexTable()
89 /*size=*/GetByteSizeRequirement(subshape), in TransferLiteralFromDeviceInternal()
125 TF_RET_CHECK(GetByteSizeRequirement(device_subshape) == in TransferLiteralToDeviceAsync()
136 /*size=*/GetByteSizeRequirement(device_subshape), source, in TransferLiteralToDeviceAsync()
145 /*size=*/GetByteSizeRequirement(device_subshape), source, in TransferLiteralToDeviceAsync()
172 int64 GenericTransferManager::GetByteSizeRequirement(const Shape& shape) const { in GetByteSizeRequirement() function in xla::GenericTransferManager
|
| D | transfer_manager.cc | 163 if (dest.size() < GetByteSizeRequirement(on_device_shape)) { in TransferArrayToDeviceAsync()
167 dest.size(), GetByteSizeRequirement(on_device_shape)); in TransferArrayToDeviceAsync()
187 if (source.size() < GetByteSizeRequirement(shape)) { in TransferArrayFromDevice()
191 source.size(), GetByteSizeRequirement(shape))); in TransferArrayFromDevice()
248 TF_RET_CHECK(GetByteSizeRequirement(device_subshape) == in WriteTupleIndexTablesAsync()
271 TF_RET_CHECK(GetByteSizeRequirement(device_buffer.on_device_shape()) == in WriteRootTupleIndexTable()
331 GetByteSizeRequirement(subshape))); in AllocateScopedShapedBuffer()
|
| D | generic_transfer_manager.h | 61 int64 GetByteSizeRequirement(const Shape& shape) const override;
|
| D | transfer_manager.h | 217 virtual int64 GetByteSizeRequirement(const Shape& shape) const = 0;
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | cpu_transfer_manager.cc | 100 int64 size = GetByteSizeRequirement(shape); in TransferLiteralToInfeed()
123 int64 tuple_element_size = GetByteSizeRequirement(tuple_element_shape); in TransferLiteralToInfeed()
181 int64 size = GetByteSizeRequirement(literal_shape); in TransferLiteralFromOutfeed()
195 TF_RET_CHECK(size == GetByteSizeRequirement(received_shape)); in TransferLiteralFromOutfeed()
209 int64 size = GetByteSizeRequirement(tuple_element_shape); in TransferLiteralFromOutfeed()
221 TF_RET_CHECK(GetByteSizeRequirement(literal_shape) == in TransferLiteralFromOutfeed()
222 GetByteSizeRequirement(received_shape)); in TransferLiteralFromOutfeed()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | gpu_transfer_manager.cc | 62 int64 tuple_element_size = GetByteSizeRequirement(literal_subshape); in TransferLiteralToInfeed()
163 GetByteSizeRequirement(shape)); in TransferLiteralFromOutfeed()
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | xla_tensor.cc | 61 client->backend().transfer_manager()->GetByteSizeRequirement(subshape); in AllocateShapedBuffer()
|
| /external/tensorflow/tensorflow/compiler/xrt/ |
| D | xrt_state.cc | 114 uint64 size = transfer_manager->GetByteSizeRequirement(subshape); in AllocateScopedShapedBuffer()
413 uint64 size = transfer_manager->GetByteSizeRequirement(subshape); in MakeTuple()
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | transfer_manager_test.cc | 45 return transfer_manager_->GetByteSizeRequirement(shape); in TransferManagerTest()
|