| /external/tensorflow/tensorflow/core/kernels/ |
| 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) +=
|
| 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
[all...] |
| depthwise_conv_grad_op.cc | 641 // First output register of 'filter_backprop'
948 Tensor* filter_backprop = nullptr; variable
[all...] |
| 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->template flat<T>().data()
591 Tensor* filter_backprop = nullptr; variable
[all...] |
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| 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);
|
