Searched refs:d_buffer (Results 1 – 5 of 5) sorted by relevance
| /external/python/pybind11/include/pybind11/ |
| D | iostream.h | 29 std::unique_ptr<char[]> d_buffer; in PYBIND11_NAMESPACE_BEGIN() local
72 d_buffer(new char[buf_size]), in PYBIND11_NAMESPACE_BEGIN()
75 setp(d_buffer.get(), d_buffer.get() + buf_size - 1); in PYBIND11_NAMESPACE_BEGIN()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | cublas_lt_matmul_thunk.cc | 33 BufferAllocation::Slice c_buffer, BufferAllocation::Slice d_buffer, in CublasLtMatmulThunk() argument
41 d_buffer_(d_buffer), in CublasLtMatmulThunk()
|
| D | matmul_utils.cc | 690 se::DeviceMemoryBase d_buffer, in DoMatmul() argument
708 se::DeviceMemory<Input> output(d_buffer); in DoMatmul()
720 se::DeviceMemoryBase d_buffer, se::DeviceMemoryBase bias_buffer, in ExecuteOnStream() argument
731 d_buffer, bias_buffer, algorithm, in ExecuteOnStream()
735 stream, a_buffer, b_buffer, c_buffer, d_buffer, bias_buffer, in ExecuteOnStream()
738 return DoMatmul<float>(stream, a_buffer, b_buffer, c_buffer, d_buffer, in ExecuteOnStream()
742 return DoMatmul<double>(stream, a_buffer, b_buffer, c_buffer, d_buffer, in ExecuteOnStream()
746 return DoMatmul<complex64>(stream, a_buffer, b_buffer, c_buffer, d_buffer, in ExecuteOnStream()
751 d_buffer, bias_buffer, algorithm, in ExecuteOnStream()
|
| D | matmul_utils.h | 136 se::DeviceMemoryBase d_buffer,
156 se::DeviceMemoryBase d_buffer,
|
| D | cublas_lt_matmul_thunk.h | 37 BufferAllocation::Slice d_buffer,
|