Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dconvolution_thunk.cc50 se::DeviceMemoryBase result_buffer = in ExecuteOnStream() local
Dgpu_conv_runner.cc496 se::DeviceMemoryBase result_buffer) { in GetGpuConvParams()
533 se::DeviceMemoryBase result_buffer, in RunGpuConv()
543 se::DeviceMemoryBase result_buffer, in RunGpuConv()
Dgpu_conv_algorithm_picker.cc128 se::DeviceMemoryBase result_buffer, se::StreamExecutor* stream_exec, in GetMIOpenAlgorithms()
Dgpu_executable.cc650 se::DeviceMemoryBase& result_buffer = p.second; in ExecuteAsyncOnStreamImpl() local
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dcpu_executable.cc190 void* result_buffer = buffer_pointers[result_slice.index()]; in ExecuteComputeFunction() local
246 se::DeviceMemoryBase& result_buffer = p.second; in CreateResultShapedBuffer() local
/external/tensorflow/tensorflow/core/lib/io/
Drandom_inputstream.cc39 char* result_buffer = &(*result)[0]; in ReadNBytes() local
/external/tensorflow/tensorflow/stream_executor/tpu/
Dtpu_executable_interface.cc112 se::DeviceMemoryBase& result_buffer = pair.second; in AllocateOutputMemoryWithInputReuse() local
/external/mesa3d/src/gallium/drivers/iris/
Diris_monitor.c36 unsigned char *result_buffer; member
/external/ComputeLibrary/src/core/NEON/kernels/arm_gemm/
Dgemm_hybrid_quantized.hpp177 …Tri *result_buffer = reinterpret_cast<Tri *>(working_int + (threadid * strategy::out_height() * _N… in execute() local
Dgemm_hybrid_indirect.hpp120 typename strategy::result_type *result_buffer; in run() local
/external/tensorflow/tensorflow/compiler/xla/pjrt/
Dpjrt_stream_executor_client.cc1699 ScopedShapedBuffer* result_buffer, in OutputBufferHelper()
2008 ScopedShapedBuffer result_buffer, in MakeOutputBuffers()
2089 ScopedShapedBuffer result_buffer = in ExecuteHelper() local
Dtfrt_cpu_pjrt_client.cc1492 void* result_buffer = buffer_pointers[result_buffer_index_]; in ExecuteHelper() local
/external/tensorflow/tensorflow/core/platform/cloud/
Dgcs_file_system.cc1557 std::vector<char> result_buffer; in GetBucketLocation() local
1576 std::vector<char>* result_buffer) { in GetBucketMetadata()
/external/libaom/libaom/av1/encoder/
Dtemporal_filter.c717 const uint16_t *count, YV12_BUFFER_CONFIG *result_buffer) { in tf_normalize_filtered_frame()
/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Ddot_handler.cc749 auto result_buffer = CreateZero(padded_result_buffer_shape, b); in PartitionBaseCase() local