| /external/tensorflow/tensorflow/core/kernels/neon/ |
| neon_depthwise_conv_op.cc | 46 OP_REQUIRES_OK(context, context->GetAttr("strides", &strides_));
47 OP_REQUIRES(context, strides_.size() == 4,
50 OP_REQUIRES(context, strides_[1] == strides_[2],
55 context, (strides_[0] == 1 && strides_[3] == 1),
88 const int32 stride = strides_[1];
189 std::vector<int32> strides_; member in class:tensorflow::NeonDepthwiseConv2dNativeOp
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| extract_image_patches_op.cc | 30 OP_REQUIRES_OK(ctx, ctx->GetAttr("strides", &strides_));
44 OP_REQUIRES(ctx, strides_.size() == num_dims,
61 ctx, strides_[batch_dim] == 1 && strides_[feature_dim] == 1,
76 OP_REQUIRES(ctx, strides_[input_dim] >= 1,
139 window_strides[i] = strides_.at(dim);
159 std::vector<int32> strides_; member in class:tensorflow::__anon38742::ExtractImagePatchesOp
|
| conv_ops.cc | 183 OP_REQUIRES_OK(ctx, ctx->GetAttr("strides", &strides_));
195 OP_REQUIRES(ctx, strides_.size() == num_dims(),
202 ctx, strides_[batch_dim] == 1 && strides_[feature_dim] == 1,
276 window_strides[i] = strides_.at(dim);
297 std::vector<int32> strides_; member in class:tensorflow::__anon38736::ConvOp
335 OP_REQUIRES_OK(ctx, ctx->GetAttr("strides", &strides_));
346 OP_REQUIRES(ctx, strides_.size() == num_dims(),
353 ctx, strides_[batch_dim] == 1 && strides_[feature_dim] == 1
450 std::vector<int32> strides_; member in class:tensorflow::__anon38736::ConvBackpropInputOp
652 std::vector<int32> strides_; member in class:tensorflow::__anon38736::ConvBackpropFilterOp
[all...] |
| /external/tensorflow/tensorflow/core/kernels/ |
| extract_image_patches_op.cc | 56 ParseAttributeVec4(context, "strides", &strides_);
77 const int stride_rows = strides_[1];
78 const int stride_cols = strides_[2];
115 std::vector<int32> strides_; member in class:tensorflow::ExtractImagePatchesOp
|
| conv_grad_input_ops.cc | 193 OP_REQUIRES_OK(context, context->GetAttr("strides", &strides_));
194 OP_REQUIRES(context, strides_.size() == 4,
198 context, (strides_[0] == 1 && strides_[3] == 1),
201 OP_REQUIRES(context, strides_[1] > 0 && strides_[2] > 0,
238 strides_, padding_, data_format_, &dims));
288 std::vector<int32> strides_; member in class:tensorflow::Conv2DFastBackpropInputOp
308 OP_REQUIRES_OK(context, context->GetAttr("strides", &strides_));
309 OP_REQUIRES(context, strides_.size() == 4
569 std::vector<int32> strides_; member in class:tensorflow::Conv2DCustomBackpropInputOp
689 std::vector<int32> strides_; member in class:tensorflow::Conv2DSlowBackpropInputOp
[all...] |
| conv_grad_filter_ops.cc | 190 OP_REQUIRES_OK(context, context->GetAttr("strides", &strides_));
191 OP_REQUIRES(context, strides_.size() == 4,
195 context, (strides_[0] == 1 && strides_[3] == 1),
198 OP_REQUIRES(context, strides_[1] > 0 && strides_[2] > 0,
235 out_backprop.shape(), strides_, padding_, data_format_, &dims));
283 std::vector<int32> strides_; member in class:tensorflow::Conv2DFastBackpropFilterOp
303 OP_REQUIRES_OK(context, context->GetAttr("strides", &strides_));
304 OP_REQUIRES(context, strides_.size() == 4
497 std::vector<int32> strides_; member in class:tensorflow::Conv2DCustomBackpropFilterOp
620 std::vector<int32> strides_; member in class:tensorflow::Conv2DSlowBackpropFilterOp
[all...] |
| conv_ops_using_gemm.cc | 433 OP_REQUIRES_OK(context, context->GetAttr("strides", &strides_));
441 OP_REQUIRES(context, strides_.size() == 4,
444 const int64 stride_n = GetTensorDim(strides_, data_format_, 'N');
445 const int64 stride_c = GetTensorDim(strides_, data_format_, 'C');
517 const int stride_rows = GetTensorDim(strides_, data_format_, 'H');
518 const int stride_cols = GetTensorDim(strides_, data_format_, 'W');
556 std::vector<int32> strides_; member in class:tensorflow::Conv2DUsingGemmOp
|
| depthwise_conv_op.cc | 263 OP_REQUIRES_OK(context, context->GetAttr("strides", &strides_));
269 OP_REQUIRES(context, strides_.size() == 4,
272 stride_ = GetTensorDim(strides_, data_format_, 'H');
273 const int64 stride_w = GetTensorDim(strides_, data_format_, 'W');
274 const int64 stride_n = GetTensorDim(strides_, data_format_, 'N');
275 const int64 stride_c = GetTensorDim(strides_, data_format_, 'C');
407 std::vector<int32> strides_; member in class:tensorflow::DepthwiseConv2dNativeOp
|
| dilation_ops.cc | 118 ParseAttributes(context, &strides_, &rates_, &padding_);
130 ParseSizes(context, strides_, rates_, padding_, &stride_rows, &stride_cols,
155 std::vector<int32> strides_; member in class:tensorflow::DilationOp
216 ParseAttributes(context, &strides_, &rates_, &padding_);
229 ParseSizes(context, strides_, rates_, padding_, &stride_rows, &stride_cols,
262 std::vector<int32> strides_; member in class:tensorflow::DilationBackpropInputOp
336 ParseAttributes(context, &strides_, &rates_, &padding_);
349 ParseSizes(context, strides_, rates_, padding_, &stride_rows, &stride_cols,
382 std::vector<int32> strides_; member in class:tensorflow::DilationBackpropFilterOp
|
| mkl_conv_ops.h | 60 std::vector<int32> strides_; member in class:tensorflow::MklDnnConvUtil
67 : context_(context), strides_(strides), padding_(pad), data_format_(fm) {}
76 int stride_rows = GetTensorDim(strides_, data_format_, 'H');
77 int stride_cols = GetTensorDim(strides_, data_format_, 'W');
340 OP_REQUIRES_OK(context, context->GetAttr("strides", &strides_));
341 int stride_n = GetTensorDim(strides_, data_format_, 'N');
342 int stride_c = GetTensorDim(strides_, data_format_, 'C');
413 MklDnnConvUtil conv_utl(context, strides_, padding_, data_format_);
552 std::vector<int32> strides_; member in class:tensorflow::MklConv2DBackpropCommonOp
|
| mkl_conv_ops.cc | 71 OP_REQUIRES_OK(context, context->GetAttr("strides", &strides_));
76 OP_REQUIRES(context, strides_.size() == 4,
80 const int64 stride_n = GetTensorDim(strides_, data_format_, 'N');
81 const int64 stride_c = GetTensorDim(strides_, data_format_, 'C');
171 const int stride_rows = GetTensorDim(strides_, data_format_, 'H');
172 const int stride_cols = GetTensorDim(strides_, data_format_, 'W');
481 std::vector<int32> strides_; member in class:tensorflow::MklConv2DOp
494 OP_REQUIRES_OK(context, context->GetAttr("strides", &strides_));
499 OP_REQUIRES(context, strides_.size() == 4,
503 const int64 stride_n = GetTensorDim(strides_, data_format_, 'N')
658 std::vector<int32> strides_; member in class:tensorflow::MklConv2DOp
[all...] |
| quantized_conv_ops.cc | 454 OP_REQUIRES_OK(context, context->GetAttr("strides", &strides_));
455 OP_REQUIRES(context, strides_.size() == 4,
458 OP_REQUIRES(context, strides_[1] == strides_[2],
463 context, (strides_[0] == 1 && strides_[3] == 1),
538 const int stride = strides_[1];
583 std::vector<int32> strides_; member in class:tensorflow::QuantizedConv2DOp
|
| depthwise_conv_grad_op.cc | 527 OP_REQUIRES_OK(context, context->GetAttr("strides", &strides_));
528 OP_REQUIRES(context, strides_.size() == 4,
537 stride_ = GetTensorDim(strides_, data_format_, 'H');
538 const int64 stride_w = GetTensorDim(strides_, data_format_, 'W');
539 const int64 stride_n = GetTensorDim(strides_, data_format_, 'N');
540 const int64 stride_c = GetTensorDim(strides_, data_format_, 'C');
587 std::vector<int32> strides_; member in class:tensorflow::DepthwiseConv2dNativeBackpropInputOp
965 std::vector<int32> strides_; member in class:tensorflow::DepthwiseConv2dNativeBackpropFilterOp
[all...] |
| conv_ops.cc | 261 OP_REQUIRES_OK(context, context->GetAttr("strides", &strides_));
272 OP_REQUIRES(context, strides_.size() == 4,
275 const int64 stride_n = GetTensorDim(strides_, data_format_, 'N');
276 const int64 stride_c = GetTensorDim(strides_, data_format_, 'C');
277 const int64 stride_h = GetTensorDim(strides_, data_format_, 'H');
278 const int64 stride_w = GetTensorDim(strides_, data_format_, 'W');
367 const int stride_rows = GetTensorDim(strides_, data_format_, 'H');
368 const int stride_cols = GetTensorDim(strides_, data_format_, 'W');
429 std::vector<int32> strides_; member in class:tensorflow::Conv2DOp
[all...] |
| mkl_conv_grad_filter_ops.cc | 70 OP_REQUIRES_OK(context, context->GetAttr("strides", &strides_));
71 int stride_n = GetTensorDim(strides_, data_format_, 'N');
72 int stride_c = GetTensorDim(strides_, data_format_, 'C');
134 input_shape, filter_shape, out_backprop_shape, strides_,
422 std::vector<int32> strides_; member in class:tensorflow::MklConv2DCustomBackpropFilterOp
|
| conv_ops_fused.cc | 631 OP_REQUIRES_OK(context, context->GetAttr("strides", &strides_));
632 OP_REQUIRES(context, strides_.size() == 4,
635 const int64 stride_n = GetTensorDim(strides_, FORMAT_NHWC, 'N');
636 const int64 stride_c = GetTensorDim(strides_, FORMAT_NHWC, 'C');
865 std::vector<int32> strides_; member in class:tensorflow::FusedResizeConv2DUsingGemmOp
[all...] |
| /external/tensorflow/tensorflow/stream_executor/ |
| dnn.cc | 476 strides_(ndims, 1) {}
485 strides_ = other.strides_;
496 port::Appendf(&strides, "%lld ", strides_[i]);
512 port::Appendf(&strides, "_s%d:%lld", i, strides_[i]);
|
| dnn.h | 653 SetDim(&strides_, DimIndex::Y, value);
657 SetDim(&strides_, DimIndex::X, value);
661 SetDim(&strides_, dim, value);
682 int64 vertical_stride() const { return GetDim(strides_, DimIndex::Y); }
683 int64 horizontal_stride() const { return GetDim(strides_, DimIndex::X); }
684 int64 stride(DimIndex dim) const { return GetDim(strides_, dim); }
687 std::vector<int64> strides() const { return strides_; }
698 std::vector<int64> strides_; member in class:perftools::gputools::dnn::PoolingDescriptor
[all...] |
| /external/tensorflow/tensorflow/core/util/ |
| mkl_util.h | 75 if (strides_) delete[] strides_;
101 strides_ = new size_t[dimension];
105 strides_[ii] = strides[ii];
162 const size_t* GetStrides() const { return strides_; }
219 // strides_
230 // sizeof(strides_)
271 strides_ = new size_t[dimension_];
276 strides_[i] = reinterpret_cast<const size_t*>(
301 strides_[i]
323 size_t* strides_ = nullptr; \/\/ Required by MKL for conversions member in class:tensorflow::MklShape
[all...] |
| /external/tensorflow/tensorflow/contrib/lite/kernels/ |
| strided_slice_test.cc | 38 strides_ = AddInput(TensorType_INT32);
58 PopulateTensor<int32>(strides_, data);
68 int strides_; member in class:tflite::__anon39279::StridedSliceOpModel
|
