| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | extract_image_patches_op.cc | 32 OP_REQUIRES_OK(ctx, ctx->GetAttr("rates", &dilations_)); in ExtractImagePatchesOp()
49 OP_REQUIRES(ctx, dilations_.size() == num_dims, in Compile()
66 ctx, dilations_[batch_dim] == 1 && dilations_[feature_dim] == 1, in Compile()
76 dilations_[input_dim])); in Compile()
80 dilations_[input_dim])); in Compile()
81 OP_REQUIRES(ctx, dilations_[input_dim] >= 1, in Compile()
84 dilations_[input_dim])); in Compile()
139 rhs_dilation[i] = dilations_.at(dim); in Compile()
157 std::vector<int32> dilations_; member in tensorflow::__anon65d92ac10111::ExtractImagePatchesOp
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | mkl_conv_ops.h | 54 std::vector<int32> dilations_; variable
64 dilations_(dilations), in context_()
92 if (dilations_.size() == 4) { in GetDilationsInMklOrder()
93 int dilations_rows = GetTensorDim(dilations_, data_format_, 'H'); in GetDilationsInMklOrder()
94 int dilations_cols = GetTensorDim(dilations_, data_format_, 'W'); in GetDilationsInMklOrder()
96 } else if (dilations_.size() == 5) { in GetDilationsInMklOrder()
97 int dilations_planes = GetTensorDim(dilations_, data_format_, '0'); in GetDilationsInMklOrder()
98 int dilations_rows = GetTensorDim(dilations_, data_format_, '1'); in GetDilationsInMklOrder()
99 int dilations_cols = GetTensorDim(dilations_, data_format_, '2'); in GetDilationsInMklOrder()
571 OP_REQUIRES_OK(context, context->GetAttr("dilations", &dilations_)); in MklConvBackpropCommonOp()
[all …]
|
| D | conv_grad_filter_ops.cc | 220 OP_REQUIRES_OK(context, context->GetAttr("dilations", &dilations_)); in Conv2DCustomBackpropFilterOp()
221 OP_REQUIRES(context, dilations_.size() == 4, in Conv2DCustomBackpropFilterOp()
224 OP_REQUIRES(context, (dilations_[0] == 1 && dilations_[3] == 1), in Conv2DCustomBackpropFilterOp()
229 OP_REQUIRES(context, (dilations_[1] == 1 && dilations_[2] == 1), in Conv2DCustomBackpropFilterOp()
402 std::vector<int32> dilations_; member in tensorflow::Conv2DCustomBackpropFilterOp
464 OP_REQUIRES_OK(context, context->GetAttr("dilations", &dilations_)); in Conv2DSlowBackpropFilterOp()
465 OP_REQUIRES(context, dilations_.size() == 4, in Conv2DSlowBackpropFilterOp()
468 int dilation_n = GetTensorDim(dilations_, data_format_, 'N'); in Conv2DSlowBackpropFilterOp()
469 int dilation_c = GetTensorDim(dilations_, data_format_, 'C'); in Conv2DSlowBackpropFilterOp()
470 int dilation_h = GetTensorDim(dilations_, data_format_, 'H'); in Conv2DSlowBackpropFilterOp()
[all …]
|
| D | conv_grad_input_ops.cc | 289 OP_REQUIRES_OK(context, context->GetAttr("dilations", &dilations_)); in Conv2DCustomBackpropInputOp()
290 OP_REQUIRES(context, dilations_.size() == 4, in Conv2DCustomBackpropInputOp()
293 OP_REQUIRES(context, (dilations_[0] == 1 && dilations_[3] == 1), in Conv2DCustomBackpropInputOp()
298 OP_REQUIRES(context, (dilations_[1] == 1 && dilations_[2] == 1), in Conv2DCustomBackpropInputOp()
536 std::vector<int32> dilations_; member in tensorflow::Conv2DCustomBackpropInputOp
601 OP_REQUIRES_OK(context, context->GetAttr("dilations", &dilations_)); in Conv2DSlowBackpropInputOp()
602 OP_REQUIRES(context, dilations_.size() == 4, in Conv2DSlowBackpropInputOp()
605 int dilation_n = GetTensorDim(dilations_, data_format_, 'N'); in Conv2DSlowBackpropInputOp()
606 int dilation_c = GetTensorDim(dilations_, data_format_, 'C'); in Conv2DSlowBackpropInputOp()
607 int dilation_h = GetTensorDim(dilations_, data_format_, 'H'); in Conv2DSlowBackpropInputOp()
[all …]
|
| D | mkl_conv_ops.cc | 862 OP_REQUIRES_OK(context, context->GetAttr("dilations", &dilations_)); in MklConvOp()
887 OP_REQUIRES(context, dilations_.size() == 4, in MklConvOp()
890 const int64 dilation_n = GetTensorDim(dilations_, data_format_, 'N'); in MklConvOp()
891 const int64 dilation_c = GetTensorDim(dilations_, data_format_, 'C'); in MklConvOp()
892 const int64 dilation_h = GetTensorDim(dilations_, data_format_, 'H'); in MklConvOp()
893 const int64 dilation_w = GetTensorDim(dilations_, data_format_, 'W'); in MklConvOp()
902 OP_REQUIRES(context, dilations_.size() == 5, in MklConvOp()
906 (GetTensorDim(dilations_, data_format_, 'N') == 1 && in MklConvOp()
907 GetTensorDim(dilations_, data_format_, 'C') == 1), in MklConvOp()
913 (GetTensorDim(dilations_, data_format_, '0') > 0 && in MklConvOp()
[all …]
|
| D | mkl_conv_grad_input_ops.cc | 367 this->data_format_, this->dilations_); in Compute()
|
| D | mkl_conv_grad_filter_ops.cc | 431 this->data_format_, this->dilations_); in Compute()
|