| /external/tensorflow/tensorflow/core/kernels/ |
| D | avgpooling_op.cc | 68 OP_REQUIRES_OK(context, context->GetAttr("ksize", &ksize_)); in AvgPoolingOp()
69 OP_REQUIRES(context, ksize_.size() == 4, in AvgPoolingOp()
77 OP_REQUIRES(context, ksize_[0] == 1 && stride_[0] == 1, in AvgPoolingOp()
81 for (int i = 0; i < ksize_.size(); ++i) { in AvgPoolingOp()
82 OP_REQUIRES(context, ksize_[i] != 0, in AvgPoolingOp()
90 ksize_, in Compute()
115 std::vector<int32> ksize_; member in tensorflow::AvgPoolingOp
141 OP_REQUIRES_OK(context, context->GetAttr("ksize", &ksize_)); in AvgPoolingOp()
142 OP_REQUIRES(context, ksize_.size() == 4, in AvgPoolingOp()
150 const int32 ksize_n = GetTensorDim(ksize_, data_format_, 'N'); in AvgPoolingOp()
[all …]
|
| D | maxpooling_op.cc | 235 OP_REQUIRES_OK(context, context->GetAttr("ksize", &ksize_)); in MaxPoolingGradOp()
236 OP_REQUIRES(context, ksize_.size() == 4, in MaxPoolingGradOp()
243 OP_REQUIRES(context, ksize_[0] == 1 && stride_[0] == 1, in MaxPoolingGradOp()
247 context, ksize_[3] == 1 && stride_[3] == 1, in MaxPoolingGradOp()
286 std::vector<int32> ksize = ksize_; in Compute()
335 std::vector<int32> ksize_; member in tensorflow::MaxPoolingGradOp
355 OP_REQUIRES_OK(context, context->GetAttr("ksize", &ksize_)); in MaxPoolingGradOp()
356 OP_REQUIRES(context, ksize_.size() == 4, in MaxPoolingGradOp()
363 const int32 ksize_n = GetTensorDim(ksize_, data_format_, 'N'); in MaxPoolingGradOp()
396 std::vector<int32> ksize = ksize_; in Compute()
[all …]
|
| D | pooling_ops_3d.cc | 140 OP_REQUIRES_OK(context, context->GetAttr("ksize", &ksize_)); in Pooling3DOp()
141 OP_REQUIRES(context, ksize_.size() == 5, in Pooling3DOp()
150 (GetTensorDim(ksize_, data_format_, 'N') == 1 && in Pooling3DOp()
155 (GetTensorDim(ksize_, data_format_, 'C') == 1 && in Pooling3DOp()
174 std::array<int64, 3> window{{GetTensorDim(ksize_, data_format_, '2'), in Compute()
175 GetTensorDim(ksize_, data_format_, '1'), in Compute()
176 GetTensorDim(ksize_, data_format_, '0')}}; in Compute()
196 std::vector<int32> ksize_; member in tensorflow::Pooling3DOp
314 OP_REQUIRES_OK(context, context->GetAttr("ksize", &ksize_)); in MaxPooling3dGradOp()
315 OP_REQUIRES(context, ksize_.size() == 5, in MaxPooling3dGradOp()
[all …]
|
| D | quantized_pooling_ops.cc | 41 OP_REQUIRES_OK(context, context->GetAttr("ksize", &ksize_)); in QuantizedAvgPoolingOp()
42 OP_REQUIRES(context, ksize_.size() == 4, in QuantizedAvgPoolingOp()
50 OP_REQUIRES(context, ksize_[0] == 1 && stride_[0] == 1, in QuantizedAvgPoolingOp()
58 ksize_, in Compute()
108 std::vector<int32> ksize_; member in tensorflow::QuantizedAvgPoolingOp
|
| D | pooling_ops_common.h | 111 OP_REQUIRES_OK(context, context->GetAttr("ksize", &ksize_)); in MaxPoolingOp()
112 OP_REQUIRES(context, ksize_.size() == 4, in MaxPoolingOp()
115 for (int i = 0; i < ksize_.size(); ++i) { in MaxPoolingOp()
116 OP_REQUIRES(context, ksize_[i] > 0, in MaxPoolingOp()
129 OP_REQUIRES(context, ksize_[0] == 1 && stride_[0] == 1, in MaxPoolingOp()
137 context, ksize_, stride_, padding_, explicit_paddings_, in Compute()
296 std::vector<int32> ksize_; variable
352 OP_REQUIRES_OK(context, context->GetAttr("ksize", &ksize_));
353 OP_REQUIRES(context, ksize_.size() == 4,
360 OP_REQUIRES(context, ksize_[0] == 1 && stride_[0] == 1,
[all …]
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | pooling_ops.cc | 62 ksize_.push_back(ksize_int[i]); in PoolingOp()
79 return ksize_; in GetKernelSize()
127 std::vector<int64> ksize_; member in tensorflow::__anon9297e9860111::PoolingOp
278 OP_REQUIRES_OK(ctx, ctx->GetAttr("ksize", &ksize_)); in MaxPoolGradOp()
296 OP_REQUIRES_OK(ctx, ctx->ConstantInputAsIntVector(3, &ksize_)); in Compile()
306 OP_REQUIRES(ctx, ksize_.size() == num_dims(), in Compile()
343 xla::MaxPool(ctx->Input(0), ksize_, stride_, xla_padding, in Compile()
359 xla::SelectAndScatter(input, select, ksize_, stride_, xla_padding, in Compile()
367 std::vector<int64> ksize_; member in tensorflow::__anon9297e9860111::MaxPoolGradOp
401 OP_REQUIRES_OK(ctx, ctx->GetAttr("ksize", &ksize_)); in AvgPoolGradOp()
[all …]
|
| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_pooling_ops_common.h | 445 OP_REQUIRES_OK(context, context->GetAttr("ksize", &this->ksize_)); in MklPoolingOpBase()
446 OP_REQUIRES(context, this->ksize_.size() == 4 || this->ksize_.size() == 5, in MklPoolingOpBase()
454 OP_REQUIRES(context, this->ksize_[0] == 1 && this->stride_[0] == 1, in MklPoolingOpBase()
457 bool is_pool2d = (this->ksize_.size() == 4); in MklPoolingOpBase()
482 if (this->ksize_.size() == 4) { in GetOutputDims()
503 pool_params->Init(context, this->ksize_, this->stride_, this->padding_, in InitMklPoolParameters()
506 pool_params->Init(context, this->ksize_, this->stride_, this->padding_, in InitMklPoolParameters()
556 if (this->ksize_.size() == 4) { in AllocateEmptyOutputTensor()
588 std::vector<int32> ksize_; variable
619 (this->ksize_.size() == 4) in ConfigureInput()
[all …]
|
| D | mkl_avgpooling_op.cc | 65 bool is_pool2d = (this->ksize_.size() == 4); in Compute()
200 bool is_pool2d = (this->ksize_.size() == 4); in Compute()
|
| D | mkl_maxpooling_op.cc | 70 bool is_pool2d = (this->ksize_.size() == 4); in Compute()
256 bool is_pool2d = (this->ksize_.size() == 4); in Compute()
|