Searched refs:filter_buf (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | cudnn_conv_runner.cc | 67 se::DeviceMemoryBase filter_buf; member
122 DeviceMemory<T> filter_buf(params.filter_buf); in RunCudnnConvImpl() local
246 input_descriptor, input_buf, filter_descriptor, filter_buf, in RunCudnnConvImpl()
257 filter_descriptor, filter_buf, output_descriptor, output_buf, in RunCudnnConvImpl()
269 convolution_descriptor, filter_descriptor, &filter_buf, in RunCudnnConvImpl()
300 filter_descriptor, filter_buf, convolution_descriptor, side_input, in RunCudnnConvImpl()
347 params.filter_buf = operand_buffers[1]; in GetCudnnConvParams()
355 params.filter_buf = operand_buffers[1]; in GetCudnnConvParams()
363 params.filter_buf = result_buffer; in GetCudnnConvParams()
381 params.filter_buf = operand_buffers[1]; in GetCudnnConvParams()
|
| /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 ()()
|