Home
last modified time | relevance | path

Searched refs:ThenBlasScal (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dfft_thunk.cc178 ->ThenBlasScal(ShapeUtil::ElementsIn(output_shape_), in ExecuteOnStream()
202 ->ThenBlasScal(ShapeUtil::ElementsIn(output_shape_), in ExecuteOnStream()
/external/tensorflow/tensorflow/core/kernels/
Dfft_ops.cc441 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/
Dstream.h845 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,
Dstream.cc2324 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