Home
last modified time | relevance | path

Searched defs:result_buffer (Results 1 – 13 of 13) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dconvolution_thunk.cc51 se::DeviceMemoryBase result_buffer = in ExecuteOnStream() local
Dgpu_conv_runner.cc460 se::DeviceMemoryBase result_buffer) { in GetGpuConvParams()
497 se::DeviceMemoryBase result_buffer, in RunGpuConv()
507 se::DeviceMemoryBase result_buffer, in RunGpuConv()
Dgpu_executable.cc465 se::DeviceMemoryBase& result_buffer = p.second; in ExecuteAsyncOnStreamImpl() local
Dgpu_conv_algorithm_picker.cc127 se::DeviceMemoryBase result_buffer, se::StreamExecutor* stream_exec, in GetMIOpenAlgorithms()
/external/tensorflow/tensorflow/core/lib/io/
Drandom_inputstream.cc39 char* result_buffer = &(*result)[0]; in ReadNBytes() local
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dcpu_executable.cc187 void* result_buffer = buffer_pointers[result_slice.index()]; in ExecuteComputeFunction() local
243 se::DeviceMemoryBase& result_buffer = p.second; in CreateResultShapedBuffer() local
/external/tensorflow/tensorflow/stream_executor/tpu/
Dtpu_executable_interface.cc100 se::DeviceMemoryBase& result_buffer = pair.second; in AllocateOutputMemoryWithInputReuse() local
/external/mesa3d/src/gallium/drivers/iris/
Diris_monitor.c36 unsigned char *result_buffer; member
/external/tensorflow/tensorflow/compiler/xla/pjrt/
Dpjrt_stream_executor_client.cc1551 ScopedShapedBuffer* result_buffer, in OutputBufferHelper()
1824 ScopedShapedBuffer result_buffer, in MakeOutputBuffers()
1898 ScopedShapedBuffer result_buffer = in ExecuteHelper() local
/external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/transforms/
Dhlo_legalize_to_lhlo.cc262 Value result_buffer = InsertDynamicAllocAndDealloc( in matchAndRewrite() local
/external/tensorflow/tensorflow/core/platform/cloud/
Dgcs_file_system.cc1558 std::vector<char> result_buffer; in GetBucketLocation() local
1577 std::vector<char>* result_buffer) { in GetBucketMetadata()
/external/libaom/libaom/av1/encoder/
Dtemporal_filter.c864 const uint16_t *count, YV12_BUFFER_CONFIG *result_buffer) { in tf_normalize_filtered_frame()
/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Ddot_handler.cc631 auto result_buffer = CreateZero(padded_result_buffer_shape, b); in PartitionBaseCase() local