| /external/tensorflow/tensorflow/core/kernels/ |
| H A D | dilation_ops.cc | 364 // The computed filter_backprop has the same dimensions as the filter:
366 Tensor* filter_backprop = nullptr; variable
368 context, context->allocate_output(0, filter.shape(), &filter_backprop));
379 filter_backprop->tensor<T, 3>());
396 typename TTypes<T, 3>::Tensor filter_backprop) {
409 filter_backprop.setZero();
441 filter_backprop(h_max, w_max, d) +=
391 operator ()(const CPUDevice& d, typename TTypes<T, 4>::ConstTensor input, typename TTypes<T, 3>::ConstTensor filter, typename TTypes<T, 4>::ConstTensor out_backprop, int stride_rows, int stride_cols, int rate_rows, int rate_cols, int pad_top, int pad_left, typename TTypes<T, 3>::Tensor filter_backprop) argument
|
| H A D | dilation_ops_gpu.cu.cc | 255 typename TTypes<T, 3>::Tensor filter_backprop) {
270 // Initialize filter_backprop with all zeros.
274 total_count, filter_backprop.data());
284 rate_rows, rate_cols, pad_top, pad_left, filter_backprop.data());
250 operator ()(const GPUDevice& d, typename TTypes<T, 4>::ConstTensor input, typename TTypes<T, 3>::ConstTensor filter, typename TTypes<T, 4>::ConstTensor out_backprop, int stride_rows, int stride_cols, int rate_rows, int rate_cols, int pad_top, int pad_left, typename TTypes<T, 3>::Tensor filter_backprop) argument
|
| H A D | conv_grad_ops_3d.cc | 297 Tensor* filter_backprop; variable
299 context->allocate_output(0, filter_shape, &filter_backprop));
302 filter_backprop->template flat<T>().setZero();
363 // Now copy the filter_backprop back to the destination.
369 filter_order, filter_rev_dims, filter_backprop->tensor<T, 5>());
813 Tensor* filter_backprop; variable
815 context->allocate_output(0, filter_shape, &filter_backprop));
842 auto c_ptr = AsDeviceMemory(filter_backprop->template flat<T>().data(),
843 filter_backprop->template flat<T>().size());
868 auto c_ptr = AsDeviceMemory(filter_backprop
[all...] |
| H A D | conv_grad_filter_ops.cc | 100 Tensor* filter_backprop, TensorFormat data_format) {
103 d, filter_backprop->tensor<T, 4>(), input.tensor<T, 4>(),
237 Tensor* filter_backprop = nullptr; variable
239 context->allocate_output(0, filter_shape, &filter_backprop));
265 filter_backprop->tensor<T, 4>(), out_backprop.tensor<T, 4>(),
278 dims.spatial_dims[1].stride, padding_, filter_backprop, data_format_);
351 Tensor* filter_backprop; variable
353 context->allocate_output(0, filter_shape, &filter_backprop));
378 filter_backprop->tensor<T, 4>(), out_backprop.tensor<T, 4>(),
438 T* filter_backprop_data = filter_backprop
97 operator ()(OpKernelContext* ctx, bool use_cudnn, bool cudnn_use_autotune, const Tensor& out_backprop, const Tensor& input, int row_stride, int col_stride, const Padding& padding, Tensor* filter_backprop, TensorFormat data_format) argument
591 Tensor* filter_backprop = nullptr; variable
631 operator ()( OpKernelContext* ctx, bool use_cudnn, bool cudnn_use_autotune, const Tensor& out_backprop, const Tensor& input, int row_dilation, int col_dilation, int row_stride, int col_stride, const Padding& padding, Tensor* filter_backprop, TensorFormat data_format) argument
[all...] |
| H A D | depthwise_conv_grad_op.cc | 641 // First output register of 'filter_backprop'
728 const T* out_backprop, const T* input, T* filter_backprop,
805 memset(filter_backprop, 0, filter_spatial_size * out_depth * sizeof(T));
812 // Load data from 'filter_backprop' into vector register.
813 auto out_block_data = filter_backprop + output_base + j;
832 filter_backprop[output_base + vectorized_size + j] +=
844 T* filter_backprop) {
847 memset(filter_backprop, 0, num_filter_backprop * sizeof(T));
877 filter_backprop[filter_backprop_offset] +=
948 Tensor* filter_backprop variable
727 operator ()(OpKernelContext* ctx, const DepthwiseArgs& args, const T* out_backprop, const T* input, T* filter_backprop, TensorFormat data_format) argument
841 DepthwiseConvBackpropFilterReference(const DepthwiseArgs& args, const T* out_backprop, const T* input, T* filter_backprop) argument
[all...] |
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| H A D | conv_ops.cc | 161 const xla::ComputationDataHandle& filter_backprop,
166 CreateExpandedFilterMask(filter_shape, builder), filter_backprop,
637 auto filter_backprop = variable
642 filter_backprop = ContractFilterForDepthwiseBackprop(
643 ctx, filter_shape, ctx->input_type(0), filter_backprop, b);
645 ctx->SetOutput(0, filter_backprop);
159 ContractFilterForDepthwiseBackprop( XlaOpKernelContext* ctx, const TensorShape& filter_shape, DataType dtype, const xla::ComputationDataHandle& filter_backprop, xla::ComputationBuilder* builder) argument
|