Searched refs:data_address (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | outfeed_thunk.cc | 55 se::DeviceMemoryBase data_address; in ExecuteOnStream() local
59 data_address = buffer_allocations.GetDeviceAddress(slice); in ExecuteOnStream()
81 data_address = in ExecuteOnStream()
90 ->ThenMemcpy((*buffer)->destination()->untyped_data(), data_address, in ExecuteOnStream()
|
| D | infeed_thunk.cc | 104 se::DeviceMemoryBase data_address = in ExecuteOnStream() local
106 void* infeed_addresses[] = {data_address.opaque(), nullptr}; in ExecuteOnStream()
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/kernels/ |
| D | trt_engine_op.cc | 309 void* data_address = GetTensorAddress(&t); in ExecuteCalibration() local
310 if (data_address == nullptr) { in ExecuteCalibration()
319 input_data.emplace(StrCat(kInputPHName, i), data_address); in ExecuteCalibration()
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | ir_emitter.cc | 339 llvm::Value* data_address = EmitBufferPointer(data_slice, data_shape); in HandleInfeed() local
344 llvm_ir::EmitTuple(GetIrArrayFor(infeed), {data_address, token_address}, &b_); in HandleInfeed()
374 llvm_ir::EmitTuple(llvm_ir::IrArray(data_address, data_shape), in HandleInfeed()
378 EmitXfeedTransfer(XfeedKind::kInfeed, data_shape, data_address)); in HandleInfeed()
|