| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | runtime_single_threaded_matmul.cc | 33 void MatMul(const void* run_options_ptr, T* out, T* lhs, T* rhs, int64_t m, in MatMul()
67 void SingleThreadedMatMulDispatch(const void* run_options_ptr, T* out, T* lhs, in SingleThreadedMatMulDispatch()
87 const void* run_options_ptr, Eigen::half* out, Eigen::half* lhs, in __xla_cpu_runtime_EigenSingleThreadedMatMulF16()
95 __xla_cpu_runtime_EigenSingleThreadedMatMulF32(const void* run_options_ptr, in __xla_cpu_runtime_EigenSingleThreadedMatMulF32()
105 __xla_cpu_runtime_EigenSingleThreadedMatMulF64(const void* run_options_ptr, in __xla_cpu_runtime_EigenSingleThreadedMatMulF64()
117 const void* run_options_ptr, std::complex<float>* out, in __xla_cpu_runtime_EigenSingleThreadedMatMulC64()
126 const void* run_options_ptr, std::complex<double>* out, in __xla_cpu_runtime_EigenSingleThreadedMatMulC128()
135 const void* run_options_ptr, tensorflow::int32* out, tensorflow::int32* lhs, in __xla_cpu_runtime_EigenSingleThreadedMatMulS32()
|
| D | runtime_matmul.cc | 37 void MatMul(const void* run_options_ptr, T* out, T* lhs, T* rhs, int64_t m, in MatMul()
75 void MatMulDispatch(const void* run_options_ptr, T* out, T* lhs, T* rhs, in MatMulDispatch()
94 const void* run_options_ptr, Eigen::half* out, Eigen::half* lhs, in __xla_cpu_runtime_EigenMatMulF16()
102 const void* run_options_ptr, float* out, float* lhs, float* rhs, int64_t m, in __xla_cpu_runtime_EigenMatMulF32()
109 const void* run_options_ptr, double* out, double* lhs, double* rhs, in __xla_cpu_runtime_EigenMatMulF64()
117 const void* run_options_ptr, std::complex<float>* out, in __xla_cpu_runtime_EigenMatMulC64()
125 const void* run_options_ptr, std::complex<double>* out, in __xla_cpu_runtime_EigenMatMulC128()
133 const void* run_options_ptr, tensorflow::int32* out, tensorflow::int32* lhs, in __xla_cpu_runtime_EigenMatMulS32()
|
| D | runtime_matmul_mkl.cc | 37 void MatMulF32(const void* run_options_ptr, float* out, float* lhs, float* rhs, in MatMulF32()
59 void MatMulF64(const void* run_options_ptr, double* out, double* lhs, in MatMulF64()
79 const void* run_options_ptr, float* out, float* lhs, float* rhs, int64 m, in __xla_cpu_runtime_MKLMatMulF32()
94 const void* run_options_ptr, double* out, double* lhs, double* rhs, int64 m, in __xla_cpu_runtime_MKLMatMulF64()
108 __xla_cpu_runtime_MKLSingleThreadedMatMulF32(const void* run_options_ptr, in __xla_cpu_runtime_MKLSingleThreadedMatMulF32()
121 __xla_cpu_runtime_MKLSingleThreadedMatMulF64(const void* run_options_ptr, in __xla_cpu_runtime_MKLSingleThreadedMatMulF64()
|
| D | runtime_matmul_mkl.h | 46 extern void __xla_cpu_runtime_MKLMatMulF32( in __xla_cpu_runtime_MKLMatMulF32()
54 extern void __xla_cpu_runtime_MKLMatMulF64( in __xla_cpu_runtime_MKLMatMulF64()
62 extern void __xla_cpu_runtime_MKLSingleThreadedMatMulF32( in __xla_cpu_runtime_MKLSingleThreadedMatMulF32()
70 extern void __xla_cpu_runtime_MKLSingleThreadedMatMulF64( in __xla_cpu_runtime_MKLSingleThreadedMatMulF64()
|
| D | runtime_single_threaded_conv2d.cc | 24 const void* run_options_ptr, Eigen::half* out, Eigen::half* lhs, in __xla_cpu_runtime_EigenSingleThreadedConvF16()
43 const void* run_options_ptr, float* out, float* lhs, float* rhs, in __xla_cpu_runtime_EigenSingleThreadedConvF32()
|
| D | runtime_conv2d.cc | 27 const void* run_options_ptr, float* out, float* lhs, float* rhs, in __xla_cpu_runtime_EigenConvF32()
47 const void* run_options_ptr, Eigen::half* out, Eigen::half* lhs, in __xla_cpu_runtime_EigenConvF16()
|
| D | runtime_single_threaded_fft.cc | 26 const void* run_options_ptr, void* out, void* operand, int32_t fft_type, in __xla_cpu_runtime_EigenSingleThreadedFft()
|
| D | runtime_fft.cc | 30 const void* run_options_ptr, void* out, void* operand, int32_t fft_type, in __xla_cpu_runtime_EigenFft()
|
| D | runtime_fork_join.cc | 63 void* result_ptr, const void* run_options_ptr, const void** params, in __xla_cpu_runtime_ParallelForkJoin()
|
| D | runtime_conv2d_mkl.cc | 155 const void* run_options_ptr, float* out, float* lhs, float* rhs, in __xla_cpu_runtime_MKLConvF32()
|
| D | cpu_runtime.cc | 249 TF_ATTRIBUTE_NO_SANITIZE_MEMORY xla::int64 __xla_cpu_runtime_TracingStart( in __xla_cpu_runtime_TracingStart()
256 TF_ATTRIBUTE_NO_SANITIZE_MEMORY void __xla_cpu_runtime_TracingEnd( in __xla_cpu_runtime_TracingEnd()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_evaluator.cc | 2622 const std::function<void(const void* run_options_ptr, T* out, T* lhs, in MatmulArray2DImpl()
|