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.cc206 .ThenBlasScal(ShapeUtil::ElementsIn(output_shape_), in ExecuteOnStream()
221 .ThenBlasScal(ShapeUtil::ElementsIn(output_shape_), in ExecuteOnStream()
251 .ThenBlasScal(ShapeUtil::ElementsIn(output_shape_), in ExecuteOnStream()
265 .ThenBlasScal(ShapeUtil::ElementsIn(output_shape_), in ExecuteOnStream()
/external/tensorflow/tensorflow/stream_executor/
Dstream.h868 Stream &ThenBlasScal(uint64 elem_count, float alpha, DeviceMemory<float> *x,
870 Stream &ThenBlasScal(uint64 elem_count, double alpha, DeviceMemory<double> *x,
872 Stream &ThenBlasScal(uint64 elem_count, float alpha,
874 Stream &ThenBlasScal(uint64 elem_count, double alpha,
876 Stream &ThenBlasScal(uint64 elem_count, std::complex<float> alpha,
878 Stream &ThenBlasScal(uint64 elem_count, std::complex<double> alpha,
Dstream.cc1908 Stream &Stream::ThenBlasScal(uint64 elem_count, float alpha, in ThenBlasScal() function in stream_executor::Stream
1916 Stream &Stream::ThenBlasScal(uint64 elem_count, double alpha, in ThenBlasScal() function in stream_executor::Stream
1924 Stream &Stream::ThenBlasScal(uint64 elem_count, float alpha, in ThenBlasScal() function in stream_executor::Stream
1932 Stream &Stream::ThenBlasScal(uint64 elem_count, double alpha, in ThenBlasScal() function in stream_executor::Stream
1940 Stream &Stream::ThenBlasScal(uint64 elem_count, std::complex<float> alpha, in ThenBlasScal() function in stream_executor::Stream
1950 Stream &Stream::ThenBlasScal(uint64 elem_count, std::complex<double> alpha, in ThenBlasScal() function in stream_executor::Stream
/external/tensorflow/tensorflow/core/kernels/
Dfft_ops.cc545 stream->ThenBlasScal(output_shape.num_elements(), alpha, &dst, 1) in DoFFTInternal()