Searched refs:ThenBlasScal (Results 1 – 4 of 4) sorted by relevance
/external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
D | fft_thunk.cc | 178 ->ThenBlasScal(ShapeUtil::ElementsIn(output_shape_), in ExecuteOnStream() 202 ->ThenBlasScal(ShapeUtil::ElementsIn(output_shape_), in ExecuteOnStream()
|
/external/tensorflow/tensorflow/core/kernels/ |
D | fft_ops.cc | 441 stream->ThenBlasScal(output_shape.num_elements(), alpha, &dst, 1) in DoFFT() 460 stream->ThenBlasScal(output_shape.num_elements(), alpha, &dst, 1) in DoFFT() 478 stream->ThenBlasScal(output_shape.num_elements(), alpha, &dst, 1) in DoFFT()
|
/external/tensorflow/tensorflow/stream_executor/ |
D | stream.h | 845 Stream &ThenBlasScal(uint64 elem_count, float alpha, DeviceMemory<float> *x, 847 Stream &ThenBlasScal(uint64 elem_count, double alpha, DeviceMemory<double> *x, 849 Stream &ThenBlasScal(uint64 elem_count, float alpha, 851 Stream &ThenBlasScal(uint64 elem_count, double alpha, 853 Stream &ThenBlasScal(uint64 elem_count, std::complex<float> alpha, 855 Stream &ThenBlasScal(uint64 elem_count, std::complex<double> alpha,
|
D | stream.cc | 2324 Stream &Stream::ThenBlasScal(uint64 elem_count, float alpha, in ThenBlasScal() function in stream_executor::Stream 2332 Stream &Stream::ThenBlasScal(uint64 elem_count, double alpha, in ThenBlasScal() function in stream_executor::Stream 2340 Stream &Stream::ThenBlasScal(uint64 elem_count, float alpha, in ThenBlasScal() function in stream_executor::Stream 2348 Stream &Stream::ThenBlasScal(uint64 elem_count, double alpha, in ThenBlasScal() function in stream_executor::Stream 2356 Stream &Stream::ThenBlasScal(uint64 elem_count, std::complex<float> alpha, in ThenBlasScal() function in stream_executor::Stream 2365 Stream &Stream::ThenBlasScal(uint64 elem_count, std::complex<double> alpha, in ThenBlasScal() function in stream_executor::Stream
|