Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/cpu/
H A Druntime_matmul.cc36 void MatMul(const void* run_options_ptr, T* out, T* lhs, T* rhs, int64_t m, in MatMul()
74 void MatMul_Batch(const void* run_options_ptr, T* out, T* lhs, T* rhs, in MatMul_Batch()
117 void MatMulDispatch(const void* run_options_ptr, T* out, T* lhs, T* rhs, in MatMulDispatch()
134 void BatchMatMulDispatch(const void* run_options_ptr, T* out, T* lhs, T* rhs, in BatchMatMulDispatch()
152 const void* run_options_ptr, Eigen::half* out, Eigen::half* lhs, in __xla_cpu_runtime_EigenMatMulF16()
160 const void* run_options_ptr, float* out, float* lhs, float* rhs, int64_t m, in __xla_cpu_runtime_EigenMatMulF32()
167 const void* run_options_ptr, double* out, double* lhs, double* rhs, in __xla_cpu_runtime_EigenMatMulF64()
175 const void* run_options_ptr, std::complex<float>* out, in __xla_cpu_runtime_EigenMatMulC64()
183 const void* run_options_ptr, std::complex<double>* out, in __xla_cpu_runtime_EigenMatMulC128()
191 const void* run_options_ptr, int32_t* out, int32_t* lhs, int32_t* rhs, in __xla_cpu_runtime_EigenMatMulS32()
[all …]
H A Druntime_single_threaded_matmul.cc32 void MatMul(const void* run_options_ptr, T* out, T* lhs, T* rhs, int64_t m, in MatMul()
66 void SingleThreadedMatMulDispatch(const void* run_options_ptr, T* out, T* lhs, in SingleThreadedMatMulDispatch()
86 const void* run_options_ptr, Eigen::half* out, Eigen::half* lhs, in __xla_cpu_runtime_EigenSingleThreadedMatMulF16()
94 __xla_cpu_runtime_EigenSingleThreadedMatMulF32(const void* run_options_ptr, in __xla_cpu_runtime_EigenSingleThreadedMatMulF32()
104 __xla_cpu_runtime_EigenSingleThreadedMatMulF64(const void* run_options_ptr, in __xla_cpu_runtime_EigenSingleThreadedMatMulF64()
116 const void* run_options_ptr, std::complex<float>* out, in __xla_cpu_runtime_EigenSingleThreadedMatMulC64()
125 const void* run_options_ptr, std::complex<double>* out, in __xla_cpu_runtime_EigenSingleThreadedMatMulC128()
133 __xla_cpu_runtime_EigenSingleThreadedMatMulS32(const void* run_options_ptr, in __xla_cpu_runtime_EigenSingleThreadedMatMulS32()
H A Druntime_matmul_mkl.cc33 void MatMulF32(const void* run_options_ptr, float* out, float* lhs, float* rhs, in MatMulF32()
55 void MatMulF64(const void* run_options_ptr, double* out, double* lhs, in MatMulF64()
75 const void* run_options_ptr, float* out, float* lhs, float* rhs, int64_t m, in __xla_cpu_runtime_MKLMatMulF32()
90 const void* run_options_ptr, double* out, double* lhs, double* rhs, in __xla_cpu_runtime_MKLMatMulF64()
105 __xla_cpu_runtime_MKLSingleThreadedMatMulF32(const void* run_options_ptr, in __xla_cpu_runtime_MKLSingleThreadedMatMulF32()
118 __xla_cpu_runtime_MKLSingleThreadedMatMulF64(const void* run_options_ptr, in __xla_cpu_runtime_MKLSingleThreadedMatMulF64()
H A Druntime_matmul_mkl.h42 extern void __xla_cpu_runtime_MKLMatMulF32( in __xla_cpu_runtime_MKLMatMulF32()
50 extern void __xla_cpu_runtime_MKLMatMulF64( in __xla_cpu_runtime_MKLMatMulF64()
58 extern void __xla_cpu_runtime_MKLSingleThreadedMatMulF32( in __xla_cpu_runtime_MKLSingleThreadedMatMulF32()
66 extern void __xla_cpu_runtime_MKLSingleThreadedMatMulF64( in __xla_cpu_runtime_MKLSingleThreadedMatMulF64()
H A Druntime_matmul_acl.cc36 int32_t MatMulF32(const void* run_options_ptr, float* out, float* lhs, in MatMulF32()
184 const void* run_options_ptr, float* out, float* lhs, float* rhs, int64_t m, in __xla_cpu_runtime_ACLMatMulF32()
195 const void* run_options_ptr, float* out, float* lhs, float* rhs, int64_t m, in __xla_cpu_runtime_ACLBatchMatMulF32()
H A Druntime_conv2d_acl.cc30 const void* run_options_ptr, float* out, float* lhs, float* rhs, in ACLDepthwiseConvImpl()
127 int32_t ACLGemmConvImpl(const void* run_options_ptr, float* out, float* lhs, in ACLGemmConvImpl()
234 const void* run_options_ptr, float* out, float* lhs, float* rhs, in __xla_cpu_runtime_ACLConv2DF32()
H A Druntime_conv2d.cc26 const void* run_options_ptr, float* out, float* lhs, float* rhs, in __xla_cpu_runtime_EigenConv2DF32()
48 const void* run_options_ptr, Eigen::half* out, Eigen::half* lhs, in __xla_cpu_runtime_EigenConv2DF16()
H A Druntime_matmul_acl.h66 extern void __xla_cpu_runtime_ACLMatMulF32( in __xla_cpu_runtime_ACLMatMulF32()
76 extern void __xla_cpu_runtime_ACLBatchMatMulF32( in __xla_cpu_runtime_ACLBatchMatMulF32()
H A Druntime_conv3d.cc26 const void* run_options_ptr, float* out, float* lhs, float* rhs, in __xla_cpu_runtime_EigenConv3DF32()
51 const void* run_options_ptr, Eigen::half* out, Eigen::half* lhs, in __xla_cpu_runtime_EigenConv3DF16()
H A Druntime_single_threaded_fft.cc22 const void* run_options_ptr, void* out, void* operand, int32_t fft_type, in __xla_cpu_runtime_EigenSingleThreadedFft()
H A Druntime_fft.cc26 const void* run_options_ptr, void* out, void* operand, int32_t fft_type, in __xla_cpu_runtime_EigenFft()
H A Druntime_fork_join.cc61 void* result_ptr, const void* run_options_ptr, const void** params, in __xla_cpu_runtime_ParallelForkJoin()
H A Dcpu_runtime.cc284 ABSL_ATTRIBUTE_NO_SANITIZE_MEMORY int64_t __xla_cpu_runtime_TracingStart( in __xla_cpu_runtime_TracingStart()
291 ABSL_ATTRIBUTE_NO_SANITIZE_MEMORY void __xla_cpu_runtime_TracingEnd( in __xla_cpu_runtime_TracingEnd()
H A Druntime_conv2d_mkl.cc155 const void* run_options_ptr, float* out, float* lhs, float* rhs, in __xla_cpu_runtime_MKLConv2DF32()
H A Druntime_conv2d_acl.h77 extern void __xla_cpu_runtime_ACLConv2DF32( in __xla_cpu_runtime_ACLConv2DF32()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dhlo_evaluator.cc3880 const std::function<void(const void* run_options_ptr, T* out, T* lhs, in MatmulArray2DImpl()