Searched refs:untyped_data (Results 1 – 12 of 12) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | generic_transfer_manager.cc | 91 literal.untyped_data(index))); in TransferLiteralFromDeviceInternal()
133 source = subliteral.untyped_data(); in TransferLiteralToDeviceAsync()
142 source = relayed_out_literal.untyped_data(); in TransferLiteralToDeviceAsync()
|
| D | hlo_evaluator.cc | 386 memcpy(result.untyped_data(), operand_literal.untyped_data(), in HandleBitcast()
|
| /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()
189 TransferArrayBufferFromOutfeed(executor, literal.untyped_data(), size)); in TransferLiteralFromOutfeed()
210 buffer_data.push_back({literal.untyped_data({i}), size}); in TransferLiteralFromOutfeed()
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | literal.cc | 192 memset(piece->untyped_data(), 0, piece->size_bytes()); in CreateFromShape()
280 static_cast<char*>(untyped_data()) + dest_linear_index * primitive_size; in CopyElementFrom()
282 static_cast<const char*>(src_literal.untyped_data()) + in CopyElementFrom()
626 char* dest_data = static_cast<char*>(result.untyped_data()); in Broadcast()
627 const char* source_data = static_cast<const char*>(untyped_data()); in Broadcast()
709 std::memcpy(new_literal.untyped_data(), untyped_data(), size_bytes()); in Transpose()
929 hash_value, Hash64(static_cast<const char*>(untyped_data(index)), in Hash()
1939 const void* LiteralBase::Piece::untyped_data() const { in untyped_data() function in xla::LiteralBase::Piece
1944 void* LiteralBase::Piece::untyped_data() { in untyped_data() function in xla::LiteralBase::Piece
2018 memcpy(untyped_data(), s.data(), s.size()); in CopyFromProto()
[all …]
|
| D | literal.h | 88 const void* untyped_data(const ShapeIndex& shape_index = {}) const;
335 void* untyped_data();
336 const void* untyped_data() const;
574 void* untyped_data(const ShapeIndex& shape_index = {});
576 using LiteralBase::untyped_data;
|
| /external/tensorflow/tensorflow/compiler/xla/service/interpreter/ |
| D | compiler.cc | 66 operand_data.push_back(literal->untyped_data()); in HandleEvaluatorCustomCall()
69 void* output_data = output.untyped_data(); in HandleEvaluatorCustomCall()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | outfeed_thunk.cc | 90 ->ThenMemcpy((*buffer)->destination()->untyped_data(), data_address, in ExecuteOnStream()
|
| D | gpu_transfer_manager.cc | 66 literal.untyped_data(index))); in TransferLiteralToInfeed()
|
| D | gpu_executable.cc | 226 literal.untyped_data(), allocation.size(), &global)); in ResolveConstantGlobals()
|
| D | ir_emitter_unnested.cc | 1708 /*source_address=*/operand->literal().untyped_data(), in BuildHostToDeviceCopyThunk()
1952 reinterpret_cast<const uint8*>(literal.untyped_data()), num_bytes); in BuildInitializerThunk()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | literal_util.cc | 94 const void* src_ptr = literal.untyped_data(); in CopyLiteralToHostTensor()
|
| /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()
|