Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Druntime_matmul.cc40 void MatMul(const void* run_options_ptr, T* out, T* lhs, T* rhs, int64 m, in MatMul() argument
43 static_cast<const xla::ExecutableRunOptions*>(run_options_ptr); in MatMul()
76 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()
92 MatMul<T, Eigen::Aligned16>(run_options_ptr, out, lhs, rhs, m, n, k, in MatMulDispatch()
100 const void* run_options_ptr, Eigen::half* out, Eigen::half* lhs, in __xla_cpu_runtime_EigenMatMulF16() argument
103 MatMulDispatch<Eigen::half>(run_options_ptr, out, lhs, rhs, m, n, k, in __xla_cpu_runtime_EigenMatMulF16()
108 const void* run_options_ptr, float* out, float* lhs, float* rhs, int64 m, in __xla_cpu_runtime_EigenMatMulF32() argument
110 MatMulDispatch<float>(run_options_ptr, out, lhs, rhs, m, n, k, transpose_lhs, in __xla_cpu_runtime_EigenMatMulF32()
115 const void* run_options_ptr, double* out, double* lhs, double* rhs, int64 m, in __xla_cpu_runtime_EigenMatMulF64() argument
[all …]
Druntime_single_threaded_matmul.cc37 void MatMul(const void* run_options_ptr, T* out, T* lhs, T* rhs, int64 m, in MatMul() argument
70 void SingleThreadedMatMulDispatch(const void* run_options_ptr, T* out, T* lhs, in SingleThreadedMatMulDispatch() argument
77 MatMul<T, Eigen::Unaligned>(run_options_ptr, out, lhs, rhs, m, n, k, in SingleThreadedMatMulDispatch()
84 MatMul<T, Eigen::Aligned16>(run_options_ptr, out, lhs, rhs, m, n, k, in SingleThreadedMatMulDispatch()
93 const void* run_options_ptr, Eigen::half* out, Eigen::half* lhs, in __xla_cpu_runtime_EigenSingleThreadedMatMulF16() argument
96 SingleThreadedMatMulDispatch<Eigen::half>(run_options_ptr, out, lhs, rhs, m, in __xla_cpu_runtime_EigenSingleThreadedMatMulF16()
101 __xla_cpu_runtime_EigenSingleThreadedMatMulF32(const void* run_options_ptr, in __xla_cpu_runtime_EigenSingleThreadedMatMulF32() argument
106 SingleThreadedMatMulDispatch<float>(run_options_ptr, out, lhs, rhs, m, n, k, in __xla_cpu_runtime_EigenSingleThreadedMatMulF32()
111 __xla_cpu_runtime_EigenSingleThreadedMatMulF64(const void* run_options_ptr, in __xla_cpu_runtime_EigenSingleThreadedMatMulF64() argument
116 SingleThreadedMatMulDispatch<double>(run_options_ptr, out, lhs, rhs, m, n, k, in __xla_cpu_runtime_EigenSingleThreadedMatMulF64()
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_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.cc28 const void* run_options_ptr, float* out, float* lhs, float* rhs, in __xla_cpu_runtime_EigenConvF32() argument
36 static_cast<const xla::ExecutableRunOptions*>(run_options_ptr); in __xla_cpu_runtime_EigenConvF32()
46 const void* run_options_ptr, Eigen::half* out, Eigen::half* lhs, in __xla_cpu_runtime_EigenConvF16() argument
55 static_cast<const xla::ExecutableRunOptions*>(run_options_ptr); in __xla_cpu_runtime_EigenConvF16()
Druntime_matmul.h30 const void* /* xla::ExecutableRunOptions* */ run_options_ptr,
36 const void* /* xla::ExecutableRunOptions* */ run_options_ptr, float* out,
42 const void* /* xla::ExecutableRunOptions* */ run_options_ptr, double* out,
Druntime_single_threaded_matmul.h30 const void* /* xla::ExecutableRunOptions* */ run_options_ptr,
36 const void* /* xla::ExecutableRunOptions* */ run_options_ptr, float* out,
42 const void* /* xla::ExecutableRunOptions* */ run_options_ptr, double* out,
Druntime_fft.cc29 const void* run_options_ptr, void* out, void* operand, int32 fft_type, in __xla_cpu_runtime_EigenFft() argument
33 static_cast<const xla::ExecutableRunOptions*>(run_options_ptr); in __xla_cpu_runtime_EigenFft()
Druntime_single_threaded_conv2d.h25 const void* /* xla::ExecutableRunOptions* */ run_options_ptr,
39 const void* /* xla::ExecutableRunOptions* */ run_options_ptr, float* out,
Druntime_conv2d.h25 const void* /* xla::ExecutableRunOptions* */ run_options_ptr, float* out,
39 const void* /* xla::ExecutableRunOptions* */ run_options_ptr,
Druntime_single_threaded_conv2d.cc26 const void* run_options_ptr, Eigen::half* out, Eigen::half* lhs, in __xla_cpu_runtime_EigenSingleThreadedConvF16() argument
44 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_fft.h24 const void* /* xla::ExecutableRunOptions* */ run_options_ptr, void* out,
Druntime_single_threaded_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.cc153 const void* run_options_ptr, float* out, float* lhs, float* rhs, in __xla_cpu_runtime_MKLConvF32() argument
165 run_options_ptr, out, lhs, rhs, input_batch, input_rows, input_cols, in __xla_cpu_runtime_MKLConvF32()
Druntime_conv2d_mkl.h25 const void* /* xla::ExecutableRunOptions* */ run_options_ptr, float* out,
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_evaluator.cc1619 const void* run_options_ptr, T* out, T* lhs, T* rhs, int64 m, int64 n, in MatmulArray2DImpl()