Searched refs:untyped_data (Results 1 – 9 of 9) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | generic_transfer_manager.cc | 92 literal->untyped_data(index))); in TransferLiteralFromDevice()
132 source = subliteral.untyped_data(); in TransferLiteralToDevice()
137 source = relayed_out_literal->untyped_data(); in TransferLiteralToDevice()
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | cpu_transfer_manager.cc | 101 return TransferBufferToInfeed(executor, size, literal.untyped_data()); in TransferLiteralToInfeed()
127 literal.untyped_data({i}))); in TransferLiteralToInfeed()
190 executor, literal->untyped_data(), size)); in TransferLiteralFromOutfeed()
220 buffer_data.push_back({empty->untyped_data(), size}); in TransferLiteralFromOutfeed()
|
| D | external_constant_pool.cc | 41 std::memcpy(raw_pointer, literal.untyped_data(), literal_size); in Insert()
|
| D | cpu_compiler.cc | 562 const void* data = instruction->literal().untyped_data(); in RunBackend()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | literal_util.cc | 36 void* dst_ptr = literal->untyped_data(); in HostTensorToLiteral()
59 const void* src_ptr = literal.untyped_data(); in CopyLiteralToHostTensor()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | gpu_transfer_manager.cc | 57 return TransferBufferToInfeed(executor, size, literal.untyped_data()); in TransferLiteralToInfeed()
84 literal.untyped_data({i}))); in TransferLiteralToInfeed()
|
| D | ir_emitter_unnested.cc | 2173 /*source_address=*/operand->literal().untyped_data(), in BuildHostToDeviceCopyThunk()
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | literal_util.cc | 168 memset(piece.untyped_data(), 0, piece.size_bytes()); in CreateFromShape()
1674 const void* Literal::Piece::untyped_data() const { in untyped_data() function in xla::Literal::Piece
1679 void* Literal::Piece::untyped_data() { in untyped_data() function in xla::Literal::Piece
1730 memcpy(untyped_data(), s.data(), s.size()); in CopyFromProto()
1732 ConvertEndianShort(reinterpret_cast<char*>(untyped_data()), s.size()); in CopyFromProto()
1739 memcpy(untyped_data(), s.data(), s.size()); in CopyFromProto()
1741 ConvertEndianShort(reinterpret_cast<char*>(untyped_data()), s.size()); in CopyFromProto()
1829 const void* Literal::untyped_data(const ShapeIndex& shape_index) const { in untyped_data() function in xla::Literal
1830 return piece(shape_index).untyped_data(); in untyped_data()
1833 void* Literal::untyped_data(const ShapeIndex& shape_index) { in untyped_data() function in xla::Literal
[all …]
|
| D | literal_util.h | 116 const void* untyped_data(const ShapeIndex& shape_index = {}) const;
117 void* untyped_data(const ShapeIndex& shape_index = {});
671 void* untyped_data();
672 const void* untyped_data() const;
|