Home
last modified time | relevance | path

Searched refs:strides_ (Results 1 – 23 of 23) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dmkl_conv_ops.h53 std::vector<int32> strides_; variable
63 strides_(strides), in context_()
75 if (strides_.size() == 4) { in GetStridesInMklOrder()
76 int stride_rows = GetTensorDim(strides_, data_format_, 'H'); in GetStridesInMklOrder()
77 int stride_cols = GetTensorDim(strides_, data_format_, 'W'); in GetStridesInMklOrder()
79 } else if (strides_.size() == 5) { in GetStridesInMklOrder()
80 int stride_planes = GetTensorDim(strides_, data_format_, '0'); in GetStridesInMklOrder()
81 int stride_rows = GetTensorDim(strides_, data_format_, '1'); in GetStridesInMklOrder()
82 int stride_cols = GetTensorDim(strides_, data_format_, '2'); in GetStridesInMklOrder()
128 if (strides_.size() == 4) { // NCHW format for Conv2D in GetInputSizeInMklOrder()
[all …]
Dextract_volume_patches_op.cc62 ParseAttributeVec5(context, "strides", &strides_); in ExtractVolumePatchesOp()
85 const int stride_planes = strides_[1]; in Compute()
86 const int stride_rows = strides_[2]; in Compute()
87 const int stride_cols = strides_[3]; in Compute()
143 std::vector<int32> strides_; member in tensorflow::ExtractVolumePatchesOp
Dextract_image_patches_op.cc56 ParseAttributeVec4(context, "strides", &strides_); in ExtractImagePatchesOp()
77 const int stride_rows = strides_[1]; in Compute()
78 const int stride_cols = strides_[2]; in Compute()
115 std::vector<int32> strides_; member in tensorflow::ExtractImagePatchesOp
Ddilation_ops.cc118 ParseAttributes(context, &strides_, &rates_, &padding_); in DilationOp()
130 ParseSizes(context, strides_, rates_, padding_, &stride_rows, &stride_cols, in Compute()
155 std::vector<int32> strides_; member in tensorflow::DilationOp
216 ParseAttributes(context, &strides_, &rates_, &padding_); in DilationBackpropInputOp()
229 ParseSizes(context, strides_, rates_, padding_, &stride_rows, &stride_cols, in Compute()
262 std::vector<int32> strides_; member in tensorflow::DilationBackpropInputOp
336 ParseAttributes(context, &strides_, &rates_, &padding_); in DilationBackpropFilterOp()
349 ParseSizes(context, strides_, rates_, padding_, &stride_rows, &stride_cols, in Compute()
382 std::vector<int32> strides_; member in tensorflow::DilationBackpropFilterOp
Dconv_grad_filter_ops.cc199 OP_REQUIRES_OK(context, context->GetAttr("strides", &strides_)); in Conv2DCustomBackpropFilterOp()
200 OP_REQUIRES(context, strides_.size() == 4, in Conv2DCustomBackpropFilterOp()
204 context, (strides_[0] == 1 && strides_[3] == 1), in Conv2DCustomBackpropFilterOp()
207 OP_REQUIRES(context, strides_[1] > 0 && strides_[2] > 0, in Conv2DCustomBackpropFilterOp()
254 strides_, padding_, data_format_, &dims)); in Compute()
403 std::vector<int32> strides_; member in tensorflow::Conv2DCustomBackpropFilterOp
452 OP_REQUIRES_OK(context, context->GetAttr("strides", &strides_)); in Conv2DSlowBackpropFilterOp()
453 int stride_n = GetTensorDim(strides_, data_format_, 'N'); in Conv2DSlowBackpropFilterOp()
454 int stride_c = GetTensorDim(strides_, data_format_, 'C'); in Conv2DSlowBackpropFilterOp()
455 int stride_h = GetTensorDim(strides_, data_format_, 'H'); in Conv2DSlowBackpropFilterOp()
[all …]
Dconv_grad_input_ops.cc277 OP_REQUIRES_OK(context, context->GetAttr("strides", &strides_)); in Conv2DCustomBackpropInputOp()
278 OP_REQUIRES(context, strides_.size() == 4, in Conv2DCustomBackpropInputOp()
282 context, (strides_[0] == 1 && strides_[3] == 1), in Conv2DCustomBackpropInputOp()
285 OP_REQUIRES(context, strides_[1] > 0 && strides_[2] > 0, in Conv2DCustomBackpropInputOp()
331 strides_, padding_, data_format_, &dims)); in Compute()
537 std::vector<int32> strides_; member in tensorflow::Conv2DCustomBackpropInputOp
586 OP_REQUIRES_OK(context, context->GetAttr("strides", &strides_)); in Conv2DSlowBackpropInputOp()
587 OP_REQUIRES(context, strides_.size() == 4, in Conv2DSlowBackpropInputOp()
590 int stride_n = GetTensorDim(strides_, data_format_, 'N'); in Conv2DSlowBackpropInputOp()
591 int stride_c = GetTensorDim(strides_, data_format_, 'C'); in Conv2DSlowBackpropInputOp()
[all …]
Dconv_ops_using_gemm.cc433 OP_REQUIRES_OK(context, context->GetAttr("strides", &strides_)); in Conv2DUsingGemmOp()
441 OP_REQUIRES(context, strides_.size() == 4, in Conv2DUsingGemmOp()
444 const int64 stride_n = GetTensorDim(strides_, data_format_, 'N'); in Conv2DUsingGemmOp()
445 const int64 stride_c = GetTensorDim(strides_, data_format_, 'C'); in Conv2DUsingGemmOp()
517 const int stride_rows = GetTensorDim(strides_, data_format_, 'H'); in Compute()
518 const int stride_cols = GetTensorDim(strides_, data_format_, 'W'); in Compute()
556 std::vector<int32> strides_; member in tensorflow::Conv2DUsingGemmOp
Dquantized_conv_ops.cc454 OP_REQUIRES_OK(context, context->GetAttr("strides", &strides_)); in QuantizedConv2DOp()
455 OP_REQUIRES(context, strides_.size() == 4, in QuantizedConv2DOp()
458 OP_REQUIRES(context, strides_[1] == strides_[2], in QuantizedConv2DOp()
463 context, (strides_[0] == 1 && strides_[3] == 1), in QuantizedConv2DOp()
538 const int stride = strides_[1]; in Compute()
583 std::vector<int32> strides_; member in tensorflow::QuantizedConv2DOp
Ddepthwise_conv_op.cc268 OP_REQUIRES_OK(context, context->GetAttr("strides", &strides_)); in DepthwiseConv2dNativeOp()
274 OP_REQUIRES(context, strides_.size() == 4, in DepthwiseConv2dNativeOp()
277 stride_ = GetTensorDim(strides_, data_format_, 'H'); in DepthwiseConv2dNativeOp()
278 const int64 stride_w = GetTensorDim(strides_, data_format_, 'W'); in DepthwiseConv2dNativeOp()
279 const int64 stride_n = GetTensorDim(strides_, data_format_, 'N'); in DepthwiseConv2dNativeOp()
280 const int64 stride_c = GetTensorDim(strides_, data_format_, 'C'); in DepthwiseConv2dNativeOp()
438 std::vector<int32> strides_; member in tensorflow::DepthwiseConv2dNativeOp
Ddepthwise_conv_grad_op.cc541 OP_REQUIRES_OK(context, context->GetAttr("strides", &strides_)); in DepthwiseConv2dNativeBackpropInputOp()
542 OP_REQUIRES(context, strides_.size() == 4, in DepthwiseConv2dNativeBackpropInputOp()
551 stride_ = GetTensorDim(strides_, data_format_, 'H'); in DepthwiseConv2dNativeBackpropInputOp()
552 const int64 stride_w = GetTensorDim(strides_, data_format_, 'W'); in DepthwiseConv2dNativeBackpropInputOp()
553 const int64 stride_n = GetTensorDim(strides_, data_format_, 'N'); in DepthwiseConv2dNativeBackpropInputOp()
554 const int64 stride_c = GetTensorDim(strides_, data_format_, 'C'); in DepthwiseConv2dNativeBackpropInputOp()
653 std::vector<int32> strides_; member in tensorflow::DepthwiseConv2dNativeBackpropInputOp
1015 OP_REQUIRES_OK(context, context->GetAttr("strides", &strides_)); in DepthwiseConv2dNativeBackpropFilterOp()
1016 OP_REQUIRES(context, strides_.size() == 4, in DepthwiseConv2dNativeBackpropFilterOp()
1025 stride_ = GetTensorDim(strides_, data_format_, 'H'); in DepthwiseConv2dNativeBackpropFilterOp()
[all …]
Dconv_ops_fused_image_transform.cc632 OP_REQUIRES_OK(context, context->GetAttr("strides", &strides_)); in FusedResizeConv2DUsingGemmOp()
633 OP_REQUIRES(context, strides_.size() == 4, in FusedResizeConv2DUsingGemmOp()
636 const int64 stride_n = GetTensorDim(strides_, FORMAT_NHWC, 'N'); in FusedResizeConv2DUsingGemmOp()
637 const int64 stride_c = GetTensorDim(strides_, FORMAT_NHWC, 'C'); in FusedResizeConv2DUsingGemmOp()
822 const int stride_rows = GetTensorDim(strides_, FORMAT_NHWC, 'H'); in Compute()
823 const int stride_cols = GetTensorDim(strides_, FORMAT_NHWC, 'W'); in Compute()
866 std::vector<int32> strides_; member in tensorflow::FusedResizeConv2DUsingGemmOp
Dmkl_conv_ops.cc422 OP_REQUIRES_OK(context, context->GetAttr("strides", &strides_)); in MklConvOp()
427 OP_REQUIRES(context, strides_.size() == 4, in MklConvOp()
431 const int64 stride_n = GetTensorDim(strides_, data_format_, 'N'); in MklConvOp()
432 const int64 stride_c = GetTensorDim(strides_, data_format_, 'C'); in MklConvOp()
522 const int stride_rows = GetTensorDim(strides_, data_format_, 'H'); in Compute()
523 const int stride_cols = GetTensorDim(strides_, data_format_, 'W'); in Compute()
834 std::vector<int32> strides_; member in tensorflow::MklConvOp
866 OP_REQUIRES_OK(context, context->GetAttr("strides", &strides_)); in MklConvOp()
871 OP_REQUIRES(context, (strides_.size() == 4 || strides_.size() == 5), in MklConvOp()
875 const int64 stride_n = GetTensorDim(strides_, data_format_, 'N'); in MklConvOp()
[all …]
Dmkl_conv_grad_filter_ops.cc376 bool is_conv2d = (this->strides_.size() == 4); in Compute()
430 MklDnnConvUtil conv_utl(context, this->strides_, this->padding_, in Compute()
659 : ((this->strides_.size() == 4) ? memory::format::hwio in GetOutputFormat()
Dmkl_conv_grad_input_ops.cc313 bool is_conv2d = (this->strides_.size() == 4); in Compute()
366 MklDnnConvUtil conv_utl(context, this->strides_, this->padding_, in Compute()
/external/tensorflow/tensorflow/core/kernels/neon/
Dneon_depthwise_conv_op.cc46 OP_REQUIRES_OK(context, context->GetAttr("strides", &strides_)); in NeonDepthwiseConv2dNativeOp()
47 OP_REQUIRES(context, strides_.size() == 4, in NeonDepthwiseConv2dNativeOp()
50 OP_REQUIRES(context, strides_[1] == strides_[2], in NeonDepthwiseConv2dNativeOp()
55 context, (strides_[0] == 1 && strides_[3] == 1), in NeonDepthwiseConv2dNativeOp()
88 const int32 stride = strides_[1]; in Compute()
189 std::vector<int32> strides_; member in tensorflow::NeonDepthwiseConv2dNativeOp
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dextract_image_patches_op.cc31 OP_REQUIRES_OK(ctx, ctx->GetAttr("strides", &strides_)); in ExtractImagePatchesOp()
45 OP_REQUIRES(ctx, strides_.size() == num_dims, in Compile()
62 ctx, strides_[batch_dim] == 1 && strides_[feature_dim] == 1, in Compile()
77 OP_REQUIRES(ctx, strides_[input_dim] >= 1, in Compile()
138 window_strides[i] = strides_.at(dim); in Compile()
158 std::vector<int32> strides_; member in tensorflow::__anon65d92ac10111::ExtractImagePatchesOp
/external/tensorflow/tensorflow/contrib/eager/python/examples/revnet/
Dops.py54 strides_ = pad_strides(strides, axis=axis)
58 x, strides_, strides_, padding="VALID", data_format=data_format)
/external/tensorflow/tensorflow/stream_executor/
Ddnn.cc508 strides_(ndims, 1) {} in PoolingDescriptor()
517 strides_ = other.strides_; in CloneFrom()
528 port::Appendf(&strides, "%lld ", strides_[i]); in ToString()
544 port::Appendf(&strides, "_s%d:%lld", i, strides_[i]); in ToShortString()
Ddnn.h689 SetDim(&strides_, DimIndex::Y, value);
693 SetDim(&strides_, DimIndex::X, value);
697 SetDim(&strides_, dim, value);
718 int64 vertical_stride() const { return GetDim(strides_, DimIndex::Y); }
719 int64 horizontal_stride() const { return GetDim(strides_, DimIndex::X); }
720 int64 stride(DimIndex dim) const { return GetDim(strides_, dim); }
723 absl::Span<const int64> strides() const { return strides_; }
734 std::vector<int64> strides_;
/external/tensorflow/tensorflow/lite/kernels/
Dstrided_slice_test.cc39 strides_ = AddInput(TensorType_INT32); in StridedSliceOpModel()
59 PopulateTensor<int32_t>(strides_, data); in SetStrides()
71 int strides_; member in tflite::__anon1c38552f0111::StridedSliceOpModel
/external/tensorflow/tensorflow/core/util/
Dmkl_util.h160 if (strides_) delete[] strides_; in ~MklShape()
186 strides_ = new size_t[dimension]; in SetTfLayout()
190 strides_[ii] = strides[ii]; in SetTfLayout()
247 const size_t* GetStrides() const { return strides_; } in GetStrides()
356 strides_ = new size_t[dimension_]; in DeSerializeMklShape()
361 strides_[i] = reinterpret_cast<const size_t*>( in DeSerializeMklShape()
386 strides_[i]; in SerializeMklShape()
408 size_t* strides_ = nullptr; // Required by MKL for conversions variable
/external/libaom/libaom/test/
Dnoise_model_test.cc534 strides_[c] = kWidth; in SetUp()
545 kWidth, kHeight, strides_, chroma_sub_, in NoiseModelUpdate()
564 int strides_[3]; member in NoiseModelUpdateTest
/external/tensorflow/tensorflow/lite/delegates/nnapi/
Dnnapi_delegate_test.cc1691 strides_ = AddConstInput(TensorType_INT32, strides_data, strides_shape); in StridedSliceOpModel()
1714 int strides_; member in tflite::__anon6d004b400111::StridedSliceOpModel