Home
last modified time | relevance | path

Searched refs:explicit_paddings_ (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dconv_grad_input_ops.h396 context->GetAttr("explicit_paddings", &explicit_paddings_));
397 OP_REQUIRES_OK(context, CheckValidPadding(padding_, explicit_paddings_,
456 explicit_paddings_, in_backprop, data_format_);
464 std::vector<int64> explicit_paddings_;
511 context->GetAttr("explicit_paddings", &explicit_paddings_));
512 OP_REQUIRES_OK(context, CheckValidPadding(padding_, explicit_paddings_,
533 explicit_paddings_, data_format_, &dims));
591 pad_top = explicit_paddings_[2];
592 pad_bottom = explicit_paddings_[3];
593 pad_left = explicit_paddings_[4];
[all …]
Ddepthwise_conv_op.cc297 context->GetAttr("explicit_paddings", &explicit_paddings_)); in DepthwiseConv2dNativeOp()
298 OP_REQUIRES_OK(context, CheckValidPadding(padding_, explicit_paddings_, in DepthwiseConv2dNativeOp()
377 GetExplicitPaddingForDim(explicit_paddings_, data_format_, 'H', &pad_top, in Compute()
379 GetExplicitPaddingForDim(explicit_paddings_, data_format_, 'W', &pad_left, in Compute()
446 stride_, stride_, padding_, explicit_paddings_, output, in Compute()
479 std::vector<int64> explicit_paddings_; member in tensorflow::DepthwiseConv2dNativeOp
Dconv_grad_filter_ops.cc299 context->GetAttr("explicit_paddings", &explicit_paddings_)); in Conv2DBackpropFilterOp()
300 OP_REQUIRES_OK(context, CheckValidPadding(padding_, explicit_paddings_, in Conv2DBackpropFilterOp()
363 explicit_paddings_, filter_backprop, data_format_); in Compute()
370 std::vector<int64> explicit_paddings_; member in tensorflow::Conv2DBackpropFilterOp
405 context->GetAttr("explicit_paddings", &explicit_paddings_)); in Conv2DCustomBackpropFilterOp()
406 OP_REQUIRES_OK(context, CheckValidPadding(padding_, explicit_paddings_, in Conv2DCustomBackpropFilterOp()
447 explicit_paddings_, data_format_, &dims)); in Compute()
461 pad_top = explicit_paddings_[2]; in Compute()
462 pad_bottom = explicit_paddings_[3]; in Compute()
463 pad_left = explicit_paddings_[4]; in Compute()
[all …]
Dmaxpooling_op.cc256 context, context->GetAttr("explicit_paddings", &explicit_paddings_)); in MaxPoolingGradOp()
257 OP_REQUIRES_OK(context, CheckValidPadding(padding_, explicit_paddings_, in MaxPoolingGradOp()
318 explicit_paddings_, in Compute()
338 std::vector<int64> explicit_paddings_; member in tensorflow::MaxPoolingGradOp
372 context, context->GetAttr("explicit_paddings", &explicit_paddings_)); in MaxPoolingGradOp()
373 OP_REQUIRES_OK(context, CheckValidPadding(padding_, explicit_paddings_, in MaxPoolingGradOp()
422 GetExplicitPaddingForDim(explicit_paddings_, data_format_, 'H', in Compute()
425 GetExplicitPaddingForDim(explicit_paddings_, data_format_, 'W', in Compute()
430 stride, padding_, explicit_paddings_, in Compute()
439 std::vector<int64> explicit_paddings_; member in tensorflow::MaxPoolingGradOp
[all …]
Ddepthwise_conv_grad_op.cc123 GetExplicitPaddingForDim(explicit_paddings_, data_format_, 'H', &pad_top, \
125 GetExplicitPaddingForDim(explicit_paddings_, data_format_, 'W', &pad_left, \
578 context->GetAttr("explicit_paddings", &explicit_paddings_)); in DepthwiseConv2dNativeBackpropInputOp()
579 OP_REQUIRES_OK(context, CheckValidPadding(padding_, explicit_paddings_, in DepthwiseConv2dNativeBackpropInputOp()
677 stride_, stride_, padding_, explicit_paddings_, in_backprop, in Compute()
696 std::vector<int64> explicit_paddings_; member in tensorflow::DepthwiseConv2dNativeBackpropInputOp
1081 context->GetAttr("explicit_paddings", &explicit_paddings_)); in DepthwiseConv2dNativeBackpropFilterOp()
1082 OP_REQUIRES_OK(context, CheckValidPadding(padding_, explicit_paddings_, in DepthwiseConv2dNativeBackpropFilterOp()
1176 padding_, explicit_paddings_, &reshaped_filter, data_format_); in Compute()
1227 std::vector<int64> explicit_paddings_; member in tensorflow::DepthwiseConv2dNativeBackpropFilterOp
Dconv_ops_fused_impl.h104 explicit_paddings_(explicit_paddings) {} in LaunchFusedConv2DWithOutputKernel()
158 static_cast<int>(explicit_paddings_[2]), in operator()
159 static_cast<int>(explicit_paddings_[3]), in operator()
160 static_cast<int>(explicit_paddings_[4]), in operator()
161 static_cast<int>(explicit_paddings_[5]), in operator()
204 const std::vector<int64>& explicit_paddings_; variable
Dpooling_ops_common.h127 context, context->GetAttr("explicit_paddings", &explicit_paddings_)); in MaxPoolingOp()
137 context, ksize_, stride_, padding_, explicit_paddings_, in Compute()
299 std::vector<int64> explicit_paddings_; variable