| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | extract_image_patches_op.cc | 39 OP_REQUIRES_OK(ctx, ctx->GetAttr("rates", &dilations_)); in ExtractImagePatchesOp()
56 OP_REQUIRES(ctx, dilations_.size() == num_dims, in Compile()
73 ctx, dilations_[batch_dim] == 1 && dilations_[feature_dim] == 1, in Compile()
83 dilations_[input_dim])); in Compile()
87 dilations_[input_dim])); in Compile()
88 OP_REQUIRES(ctx, dilations_[input_dim] >= 1, in Compile()
91 dilations_[input_dim])); in Compile()
145 rhs_dilation[i] = dilations_.at(dim); in Compile()
173 std::vector<int32> dilations_; member in tensorflow::__anon53c893df0111::ExtractImagePatchesOp
|
| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_conv_ops.h | 60 std::vector<int32> dilations_; variable
70 dilations_(dilations), in context_()
98 if (dilations_.size() == 4) { in GetDilationsInMklOrder()
99 int dilations_rows = GetTensorDim(dilations_, data_format_, 'H'); in GetDilationsInMklOrder()
100 int dilations_cols = GetTensorDim(dilations_, data_format_, 'W'); in GetDilationsInMklOrder()
102 } else if (dilations_.size() == 5) { in GetDilationsInMklOrder()
103 int dilations_planes = GetTensorDim(dilations_, data_format_, '0'); in GetDilationsInMklOrder()
104 int dilations_rows = GetTensorDim(dilations_, data_format_, '1'); in GetDilationsInMklOrder()
105 int dilations_cols = GetTensorDim(dilations_, data_format_, '2'); in GetDilationsInMklOrder()
578 OP_REQUIRES_OK(context, context->GetAttr("dilations", &dilations_)); in MklConvBackpropCommonOp()
[all …]
|
| D | mkl_conv_ops.cc | 422 OP_REQUIRES_OK(context, context->GetAttr("dilations", &dilations_)); in MklConvOp()
464 OP_REQUIRES(context, dilations_.size() == 4, in MklConvOp()
467 const int64 dilation_n = GetTensorDim(dilations_, data_format_, 'N'); in MklConvOp()
468 const int64 dilation_c = GetTensorDim(dilations_, data_format_, 'C'); in MklConvOp()
469 const int64 dilation_h = GetTensorDim(dilations_, data_format_, 'H'); in MklConvOp()
470 const int64 dilation_w = GetTensorDim(dilations_, data_format_, 'W'); in MklConvOp()
479 OP_REQUIRES(context, dilations_.size() == 5, in MklConvOp()
483 (GetTensorDim(dilations_, data_format_, 'N') == 1 && in MklConvOp()
484 GetTensorDim(dilations_, data_format_, 'C') == 1), in MklConvOp()
490 (GetTensorDim(dilations_, data_format_, '0') > 0 && in MklConvOp()
[all …]
|
| D | mkl_conv_grad_input_ops.cc | 376 this->data_format_, this->dilations_); in Compute()
|
| D | mkl_conv_grad_filter_ops.cc | 424 this->data_format_, this->dilations_); in Compute()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | conv_grad_input_ops.h | 378 OP_REQUIRES_OK(context, context->GetAttr("dilations", &dilations_));
379 OP_REQUIRES(context, dilations_.size() == 4,
382 int dilation_n = GetTensorDim(dilations_, data_format_, 'N');
383 int dilation_c = GetTensorDim(dilations_, data_format_, 'C');
384 int dilation_h = GetTensorDim(dilations_, data_format_, 'H');
385 int dilation_w = GetTensorDim(dilations_, data_format_, 'W');
443 const int dilation_rows = GetTensorDim(dilations_, data_format_, 'H');
444 const int dilation_cols = GetTensorDim(dilations_, data_format_, 'W');
460 std::vector<int32> dilations_;
497 OP_REQUIRES_OK(context, context->GetAttr("dilations", &dilations_));
[all …]
|
| D | conv_grad_filter_ops.cc | 281 OP_REQUIRES_OK(context, context->GetAttr("dilations", &dilations_)); in Conv2DBackpropFilterOp()
282 OP_REQUIRES(context, dilations_.size() == 4, in Conv2DBackpropFilterOp()
285 int dilation_n = GetTensorDim(dilations_, data_format_, 'N'); in Conv2DBackpropFilterOp()
286 int dilation_c = GetTensorDim(dilations_, data_format_, 'C'); in Conv2DBackpropFilterOp()
287 int dilation_h = GetTensorDim(dilations_, data_format_, 'H'); in Conv2DBackpropFilterOp()
288 int dilation_w = GetTensorDim(dilations_, data_format_, 'W'); in Conv2DBackpropFilterOp()
351 const int dilation_rows = GetTensorDim(dilations_, data_format_, 'H'); in Compute()
352 const int dilation_cols = GetTensorDim(dilations_, data_format_, 'W'); in Compute()
367 std::vector<int32> dilations_; member in tensorflow::Conv2DBackpropFilterOp
408 OP_REQUIRES_OK(context, context->GetAttr("dilations", &dilations_)); in Conv2DCustomBackpropFilterOp()
[all …]
|