| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | generic_transfer_manager.cc | 77 /*host_dst=*/literal.untyped_data(index), in TransferLiteralFromDevice()
125 source = subliteral.untyped_data(); in TransferLiteralToDeviceAsync()
134 source = relayed_out_literal.untyped_data(); in TransferLiteralToDeviceAsync()
|
| D | hlo_evaluator.cc | 1039 memcpy(result.untyped_data(), operand_literal.untyped_data(), in HandleBitcast()
1075 memcpy(result.untyped_data(), operand_literal.untyped_data(), in HandleSetDimensionSize()
|
| D | hlo_parser.cc | 3681 uint32_t* raw_data = static_cast<uint32_t*>(literal->untyped_data()); in ParseDenseLiteral()
3685 uint8_t* raw_data_int8 = static_cast<uint8_t*>(literal->untyped_data()); in ParseDenseLiteral()
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | cpu_xfeed.cc | 184 return TransferBufferToInfeed(device_ordinal, size, literal.untyped_data()); in TransferLiteralToInfeedOnCpu()
210 tuple_element_size, literal.untyped_data({i}))); in TransferLiteralToInfeedOnCpu()
234 device_ordinal, literal.untyped_data(), size)); in TransferLiteralFromOutfeedOnCpu()
257 buffer_data.push_back({literal.untyped_data({i}), size}); in TransferLiteralFromOutfeedOnCpu()
|
| /external/tensorflow/tensorflow/compiler/xla/service/interpreter/ |
| D | compiler.cc | 68 operand_data.push_back(literal->untyped_data()); in HandleEvaluatorCustomCall()
71 void* output_data = output.untyped_data(); in HandleEvaluatorCustomCall()
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | literal.cc | 316 memset(piece->untyped_data(), 0, piece->size_bytes()); in CreateFromShape()
446 static_cast<char*>(untyped_data()) + dest_linear_index * primitive_size; in CopyElementFrom()
448 static_cast<const char*>(src_literal.untyped_data()) + in CopyElementFrom()
932 char* dest_data = static_cast<char*>(result.untyped_data()); in Broadcast()
933 const char* source_data = static_cast<const char*>(untyped_data()); in Broadcast()
1024 std::memcpy(new_literal.untyped_data(), untyped_data(), size_bytes()); in Transpose()
2355 const void* LiteralBase::Piece::untyped_data() const { in untyped_data() function in xla::LiteralBase::Piece
2360 void* LiteralBase::Piece::untyped_data() { in untyped_data() function in xla::LiteralBase::Piece
2418 memcpy(untyped_data(), s.data(), s.size()); in CopyFromProto()
2420 ConvertEndianShort(reinterpret_cast<char*>(untyped_data()), s.size()); in CopyFromProto()
[all …]
|
| D | literal.h | 83 const void* untyped_data(const ShapeIndex& shape_index = {}) const;
289 static_cast<const char*>(literal.untyped_data(index)), in Hash()
442 void* untyped_data();
443 const void* untyped_data() const;
797 void* untyped_data(const ShapeIndex& shape_index = {});
799 using LiteralBase::untyped_data;
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | outfeed_thunk.cc | 92 .ThenMemcpy(buffer->destination()->untyped_data(), data_address, in ExecuteOnStream()
|
| D | infeed_manager.cc | 74 literal.untyped_data(leaf.first))); in TransferLiteralToInfeed()
|
| D | ir_emitter_nested.cc | 239 auto base = static_cast<const uint8_t*>(literal.untyped_data()); in EmitConstants()
|
| D | jitrt_custom_calls.cc | 1074 auto* dest_address = buffer->destination()->untyped_data(); in operator ()()
|
| /external/tensorflow/tensorflow/compiler/xla/pjrt/ |
| D | host_callback_test.cc | 81 std::memcpy(chunk.data(), literal.untyped_data(), literal.size_bytes()); in TEST()
|
| D | tfrt_cpu_pjrt_client.cc | 623 std::memcpy(b->data(), literal.untyped_data(), b->size()); in BufferFromHostLiteral()
639 std::memcpy(b->data(), slice.untyped_data(), slice.size_bytes()); in BufferFromHostLiteral()
945 std::memcpy(literal->untyped_data(), b->data(), b->size()); in ToLiteral()
952 std::memcpy(literal->untyped_data({i}), b->data(), b->size()); in ToLiteral()
981 std::memcpy(literal->untyped_data(), b->data(), b->size()); in ToLiteral()
988 std::memcpy(literal->untyped_data({i}), b->data(), b->size()); in ToLiteral()
1035 literal_pointer->untyped_data(), in CopyToDevice()
|
| D | pjrt_stream_executor_client.cc | 1497 literal_pointer->untyped_data(), in CopyToDevice()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | literal_util.cc | 105 const void* src_ptr = literal.untyped_data(); in CopyLiteralToHostTensor()
|
| /external/tensorflow/tensorflow/compiler/xla/python/tpu_driver/client/ |
| D | tpu_client.cc | 202 driver->TransferToDevice(leaves[0].untyped_data(), handle, {}); in FromLiterals()
224 return driver->TransferToDevice(leaf.untyped_data(), handle, {}); in FromLiterals()
323 device_buffer_->handle.get(), host_value->value->untyped_data(), in CopyToHostAsync()
330 host_value->value->untyped_data(xla::ShapeIndex({i})), events)); in CopyToHostAsync()
|
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| D | types.cc | 267 ByteStridesForShape(m.shape()), m.untyped_data(), in LiteralToPython()
|
| /external/tensorflow/tensorflow/compiler/xla/stream_executor/tpu/ |
| D | c_api_conversions.cc | 370 const_cast<void*>(literal.untyped_data(shapes[i].index))); in ToC()
|
| /external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/ |
| D | llvm_util.cc | 259 const char* data = static_cast<const char*>(literal.untyped_data()); in ConvertLiteralToIrConstant()
|