Home
last modified time | relevance | path

Searched refs:run_options_ptr (Results 1 – 21 of 21) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Druntime_matmul.cc37 void MatMul(const void* run_options_ptr, T* out, T* lhs, T* rhs, in MatMul() argument
41 static_cast<const xla::ExecutableRunOptions*>(run_options_ptr); in MatMul()
75 void MatMulDispatch(const void* run_options_ptr, T* out, T* lhs, T* rhs, in MatMulDispatch() argument
83 MatMul<T, Eigen::Unaligned>(run_options_ptr, out, lhs, rhs, m, n, k, in MatMulDispatch()
88 MatMul<T, Eigen::Aligned16>(run_options_ptr, out, lhs, rhs, m, n, k, in MatMulDispatch()
95 const void* run_options_ptr, Eigen::half* out, Eigen::half* lhs, in __xla_cpu_runtime_EigenMatMulF16() argument
99 MatMulDispatch<Eigen::half>(run_options_ptr, out, lhs, rhs, m, n, k, in __xla_cpu_runtime_EigenMatMulF16()
104 const void* run_options_ptr, float* out, float* lhs, float* rhs, in __xla_cpu_runtime_EigenMatMulF32() argument
107 MatMulDispatch<float>(run_options_ptr, out, lhs, rhs, m, n, k, transpose_lhs, in __xla_cpu_runtime_EigenMatMulF32()
112 const void* run_options_ptr, double* out, double* lhs, double* rhs, in __xla_cpu_runtime_EigenMatMulF64() argument
[all …]
Druntime_single_threaded_matmul.cc33 void MatMul(const void* run_options_ptr, T* out, T* lhs, T* rhs, in MatMul() argument
67 void SingleThreadedMatMulDispatch(const void* run_options_ptr, T* out, T* lhs, in SingleThreadedMatMulDispatch() argument
76 MatMul<T, Eigen::Unaligned>(run_options_ptr, out, lhs, rhs, m, n, k, in SingleThreadedMatMulDispatch()
80 MatMul<T, Eigen::Aligned16>(run_options_ptr, out, lhs, rhs, m, n, k, in SingleThreadedMatMulDispatch()
88 const void* run_options_ptr, Eigen::half* out, Eigen::half* lhs, in __xla_cpu_runtime_EigenSingleThreadedMatMulF16() argument
92 SingleThreadedMatMulDispatch<Eigen::half>(run_options_ptr, out, lhs, rhs, m, in __xla_cpu_runtime_EigenSingleThreadedMatMulF16()
98 const void* run_options_ptr, float* out, float* lhs, float* rhs, in __xla_cpu_runtime_EigenSingleThreadedMatMulF32() argument
101 SingleThreadedMatMulDispatch<float>(run_options_ptr, out, lhs, rhs, m, n, k, in __xla_cpu_runtime_EigenSingleThreadedMatMulF32()
107 const void* run_options_ptr, double* out, double* lhs, double* rhs, in __xla_cpu_runtime_EigenSingleThreadedMatMulF64() argument
110 SingleThreadedMatMulDispatch<double>(run_options_ptr, out, lhs, rhs, m, n, k, in __xla_cpu_runtime_EigenSingleThreadedMatMulF64()
[all …]
Druntime_matmul_mkl.h25 const void* /* xla::ExecutableRunOptions* */ run_options_ptr, float* out,
30 const void* /* xla::ExecutableRunOptions* */ run_options_ptr, double* out,
35 const void* /* xla::ExecutableRunOptions* */ run_options_ptr, float* out,
40 const void* /* xla::ExecutableRunOptions* */ run_options_ptr, double* out,
47 const void* /* xla::ExecutableRunOptions* */ run_options_ptr, float* out, in __xla_cpu_runtime_MKLMatMulF32()
56 const void* /* xla::ExecutableRunOptions* */ run_options_ptr, double* out, in __xla_cpu_runtime_MKLMatMulF64()
65 const void* /* xla::ExecutableRunOptions* */ run_options_ptr, float* out, in __xla_cpu_runtime_MKLSingleThreadedMatMulF32()
74 const void* /* xla::ExecutableRunOptions* */ run_options_ptr, double* out, in __xla_cpu_runtime_MKLSingleThreadedMatMulF64()
Druntime_matmul_mkl.cc37 void MatMulF32(const void* run_options_ptr, float* out, float* lhs, float* rhs, in MatMulF32() argument
59 void MatMulF64(const void* run_options_ptr, double* out, double* lhs, in MatMulF64() argument
79 const void* run_options_ptr, float* out, float* lhs, float* rhs, int64 m, in __xla_cpu_runtime_MKLMatMulF32() argument
82 static_cast<const xla::ExecutableRunOptions*>(run_options_ptr); in __xla_cpu_runtime_MKLMatMulF32()
94 const void* run_options_ptr, double* out, double* lhs, double* rhs, int64 m, in __xla_cpu_runtime_MKLMatMulF64() argument
97 static_cast<const xla::ExecutableRunOptions*>(run_options_ptr); in __xla_cpu_runtime_MKLMatMulF64()
108 __xla_cpu_runtime_MKLSingleThreadedMatMulF32(const void* run_options_ptr, in __xla_cpu_runtime_MKLSingleThreadedMatMulF32() argument
121 __xla_cpu_runtime_MKLSingleThreadedMatMulF64(const void* run_options_ptr, in __xla_cpu_runtime_MKLSingleThreadedMatMulF64() argument
Druntime_matmul.h32 const void* /* xla::ExecutableRunOptions* */ run_options_ptr,
38 const void* /* xla::ExecutableRunOptions* */ run_options_ptr, float* out,
44 const void* /* xla::ExecutableRunOptions* */ run_options_ptr, double* out,
50 const void* run_options_ptr, std::complex<float>* out,
56 const void* run_options_ptr, std::complex<double>* out,
62 const void* /* xla::ExecutableRunOptions* */ run_options_ptr,
Druntime_single_threaded_matmul.h32 const void* /* xla::ExecutableRunOptions* */ run_options_ptr,
38 const void* /* xla::ExecutableRunOptions* */ run_options_ptr, float* out,
44 const void* /* xla::ExecutableRunOptions* */ run_options_ptr, double* out,
50 const void* /* xla::ExecutableRunOptions* */ run_options_ptr,
57 const void* /* xla::ExecutableRunOptions* */ run_options_ptr,
64 const void* /* xla::ExecutableRunOptions* */ run_options_ptr,
Druntime_fork_join.cc63 void* result_ptr, const void* run_options_ptr, const void** params, in __xla_cpu_runtime_ParallelForkJoin() argument
75 static_cast<const xla::ExecutableRunOptions*>(run_options_ptr); in __xla_cpu_runtime_ParallelForkJoin()
89 [i, function, result_ptr, run_options_ptr, buffer_table, prof_counters, in __xla_cpu_runtime_ParallelForkJoin()
91 function(result_ptr, run_options_ptr, nullptr, buffer_table, in __xla_cpu_runtime_ParallelForkJoin()
99 function(result_ptr, run_options_ptr, params, buffer_table, &partitions[0], in __xla_cpu_runtime_ParallelForkJoin()
Druntime_conv2d.cc27 const void* run_options_ptr, float* out, float* lhs, float* rhs, in __xla_cpu_runtime_EigenConvF32() argument
39 static_cast<const xla::ExecutableRunOptions*>(run_options_ptr); in __xla_cpu_runtime_EigenConvF32()
50 const void* run_options_ptr, Eigen::half* out, Eigen::half* lhs, in __xla_cpu_runtime_EigenConvF16() argument
63 static_cast<const xla::ExecutableRunOptions*>(run_options_ptr); in __xla_cpu_runtime_EigenConvF16()
Druntime_fft.cc30 const void* run_options_ptr, void* out, void* operand, int32 fft_type, in __xla_cpu_runtime_EigenFft() argument
34 static_cast<const xla::ExecutableRunOptions*>(run_options_ptr); in __xla_cpu_runtime_EigenFft()
Druntime_conv2d.h25 const void* /* xla::ExecutableRunOptions* */ run_options_ptr, float* out,
39 const void* /* xla::ExecutableRunOptions* */ run_options_ptr,
Druntime_single_threaded_conv2d.h25 const void* /* xla::ExecutableRunOptions* */ run_options_ptr,
39 const void* /* xla::ExecutableRunOptions* */ run_options_ptr, float* out,
Dcpu_runtime.h98 const void* /* xla::ExecutableRunOptions* */ run_options_ptr,
101 const void* /* xla::ExecutableRunOptions* */ run_options_ptr,
Druntime_single_threaded_conv2d.cc24 const void* run_options_ptr, Eigen::half* out, Eigen::half* lhs, in __xla_cpu_runtime_EigenSingleThreadedConvF16() argument
46 const void* run_options_ptr, float* out, float* lhs, float* rhs, in __xla_cpu_runtime_EigenSingleThreadedConvF32() argument
Druntime_fork_join.h26 void* result_ptr, const void* run_options_ptr, const void** params,
Druntime_single_threaded_fft.h24 const void* /* xla::ExecutableRunOptions* */ run_options_ptr, void* out,
Druntime_fft.h24 const void* /* xla::ExecutableRunOptions* */ run_options_ptr, void* out,
Druntime_single_threaded_fft.cc26 const void* run_options_ptr, void* out, void* operand, int32 fft_type, in __xla_cpu_runtime_EigenSingleThreadedFft() argument
Druntime_conv2d_mkl.h25 const void* /* xla::ExecutableRunOptions* */ run_options_ptr, float* out,
Druntime_conv2d_mkl.cc155 const void* run_options_ptr, float* out, float* lhs, float* rhs, in __xla_cpu_runtime_MKLConvF32() argument
167 run_options_ptr, out, lhs, rhs, input_batch, input_rows, input_cols, in __xla_cpu_runtime_MKLConvF32()
Dcpu_runtime.cc216 const void* /* xla::ExecutableRunOptions* */ run_options_ptr, in __xla_cpu_runtime_TracingStart()
223 const void* /* xla::ExecutableRunOptions* */ run_options_ptr, in __xla_cpu_runtime_TracingEnd()
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_evaluator.cc2559 const void* run_options_ptr, T* out, T* lhs, T* rhs, int64 m, int64 n, in MatmulArray2DImpl()