Home
last modified time | relevance | path

Searched defs:src_mem (Results 1 – 16 of 16) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/host/
H A Dhost_gpu_executor.cc103 void* src_mem = const_cast<void*>(gpu_src.opaque()); in Memcpy() local
124 void* src_mem = const_cast<void*>(gpu_src.opaque()); in MemcpyDeviceToDevice() local
/aosp_15_r20/external/mesa3d/src/gallium/frontends/clover/api/
H A Dtransfer.cpp183 buffer &src_mem, in validate_copy()
555 auto &src_mem = obj<buffer>(d_src_mem); in clEnqueueCopyBuffer() local
592 auto &src_mem = obj<buffer>(d_src_mem); in clEnqueueCopyBufferRect() local
802 auto &src_mem = obj<buffer>(d_src_mem); in clEnqueueCopyBufferToImage() local
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/mkl/
H A Dmkl_layer_norm_op.cc73 auto src_mem = memory(src_md, cpu_engine, src_buf); in Compute() local
H A Dmkl_eltwise_activation_base_op.h118 std::shared_ptr<memory> src_mem; member
H A Dmkl_softmax_op.cc96 std::shared_ptr<memory> src_mem; member
H A Dmkl_fused_batch_norm_op.cc167 std::shared_ptr<dnnl::memory> src_mem; member
511 std::shared_ptr<dnnl::memory> src_mem; member
H A Dmkl_relu_op.cc109 std::shared_ptr<memory> src_mem; member
296 std::shared_ptr<memory> src_mem; member
H A Dmkl_slice_op.cc220 std::shared_ptr<dnnl::memory> src_mem; member
H A Dmkl_quantize_op.cc112 std::shared_ptr<dnnl::memory> src_mem; member
H A Dmkl_conv_grad_filter_ops.cc153 std::shared_ptr<dnnl::memory> src_mem; member
H A Dmkl_matmul_ops_common.h171 std::shared_ptr<dnnl::memory> src_mem; member
H A Dmkl_concat_op.cc354 std::shared_ptr<dnnl::memory> src_mem( in Setup() local
H A Dmkl_pooling_ops_common.h120 std::shared_ptr<dnnl::memory> src_mem; member
H A Dmkl_conv_ops.cc219 std::shared_ptr<dnnl::memory> src_mem; member
/aosp_15_r20/external/mesa3d/src/amd/common/
H A Dac_parse_ib.c526 bool src_mem = (src_sel == V_501_SRC_ADDR && G_415_SAS(command) == V_415_MEMORY) || in ac_parse_packet3() local
/aosp_15_r20/external/tensorflow/tensorflow/stream_executor/rocm/
H A Drocm_blas.cc679 DeviceMemoryBase src_mem = DeviceMemoryBase(src_ptr, cur_stride_size); in ReorganizeMemory() local
696 DeviceMemoryBase src_mem = DeviceMemoryBase(src_ptr, cur_stride_size); in ReorganizeMemory() local