Searched refs:filter_buf (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | gpu_conv_runner.cc | 80 DeviceMemory<ElementType> filter_buf, in RunGpuConvForward() argument
90 params.config.filter_descriptor, filter_buf, params.config.conv_desc, in RunGpuConvForward()
100 DeviceMemory<ElementType> filter_buf, in RunGpuConvForwardActivation() argument
131 filter_buf, params.config.conv_desc, side_input, in RunGpuConvForwardActivation()
153 DeviceMemory<ElementType> filter_buf, in RunGpuConvInternalImpl() argument
159 input_buf, filter_buf, output_buf, algorithm); in RunGpuConvInternalImpl()
167 params.config.filter_descriptor, filter_buf, in RunGpuConvInternalImpl()
181 params.config.filter_descriptor, &filter_buf, scratch_allocator, in RunGpuConvInternalImpl()
186 params, scratch_allocator, stream, options, input_buf, filter_buf, in RunGpuConvInternalImpl()
201 DeviceMemory<ElementType> filter_buf, in RunGpuConvInternalImpl() argument
[all …]
|
| D | gpu_conv_runner.h | 90 se::DeviceMemoryBase filter_buf; member
|
| D | gpu_conv_algorithm_picker.cc | 144 params.config.filter_descriptor, params.filter_buf, in GetMIOpenAlgorithms()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | deep_conv2d.cc | 142 void operator()(const Conv2DArgs& args, const T* filter_in, T* filter_buf) { in operator ()()
154 Eigen::internal::pstoreu<T>(filter_buf + d * kPacketSize, v); in operator ()()
160 filter_buf[buf_scalar_base + d] = in operator ()()
288 const T* transform_matrix, T* out_buffer, T* filter_buf, in operator ()()
315 memset(filter_buf, 0, sizeof(T) * filter_buf_size); in operator ()()
350 filter_buf + buf_index); in operator ()()
361 shard_rows, shard_cols, filter_buf, in operator ()()
451 T* filter_buf = filter_transform_buffer.template flat<T>().data(); in operator ()() local
472 filter_buf, filter_out); in operator ()()
478 filter_buf, filter_out); in operator ()()
|