| /external/tensorflow/tensorflow/lite/kernels/ |
| strided_slice_test.cc | 39 strides_ = AddInput(TensorType_INT32);
59 PopulateTensor<int32_t>(strides_, data);
71 int strides_; member in class:tflite::__anon46104::StridedSliceOpModel
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| extract_image_patches_op.cc | 31 OP_REQUIRES_OK(ctx, ctx->GetAttr("strides", &strides_));
45 OP_REQUIRES(ctx, strides_.size() == num_dims,
62 ctx, strides_[batch_dim] == 1 && strides_[feature_dim] == 1,
77 OP_REQUIRES(ctx, strides_[input_dim] >= 1,
138 window_strides[i] = strides_.at(dim);
158 std::vector<int32> strides_; member in class:tensorflow::__anon44134::ExtractImagePatchesOp
|
| /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
|
| extract_volume_patches_op.cc | 62 ParseAttributeVec5(context, "strides", &strides_);
85 const int stride_planes = strides_[1];
86 const int stride_rows = strides_[2];
87 const int stride_cols = strides_[3];
143 std::vector<int32> strides_; member in class:tensorflow::ExtractVolumePatchesOp
|
| 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
|
| 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 | 268 OP_REQUIRES_OK(context, context->GetAttr("strides", &strides_));
274 OP_REQUIRES(context, strides_.size() == 4,
277 stride_ = GetTensorDim(strides_, data_format_, 'H');
278 const int64 stride_w = GetTensorDim(strides_, data_format_, 'W');
279 const int64 stride_n = GetTensorDim(strides_, data_format_, 'N');
280 const int64 stride_c = GetTensorDim(strides_, data_format_, 'C');
438 std::vector<int32> strides_; member in class:tensorflow::DepthwiseConv2dNativeOp
|
| mkl_conv_ops.h | 53 std::vector<int32> strides_; member in class:tensorflow::MklDnnConvUtil
63 strides_(strides),
75 if (strides_.size() == 4) {
76 int stride_rows = GetTensorDim(strides_, data_format_, 'H');
77 int stride_cols = GetTensorDim(strides_, data_format_, 'W');
79 } else if (strides_.size() == 5) {
80 int stride_planes = GetTensorDim(strides_, data_format_, '0');
81 int stride_rows = GetTensorDim(strides_, data_format_, '1');
82 int stride_cols = GetTensorDim(strides_, data_format_, '2');
128 if (strides_.size() == 4) { // NCHW format for Conv2
602 std::vector<int32> strides_; member in class:tensorflow::MklConvBackpropCommonOp
[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
|
| conv_grad_filter_ops.cc | 199 OP_REQUIRES_OK(context, context->GetAttr("strides", &strides_));
200 OP_REQUIRES(context, strides_.size() == 4,
204 context, (strides_[0] == 1 && strides_[3] == 1),
207 OP_REQUIRES(context, strides_[1] > 0 && strides_[2] > 0,
254 strides_, padding_, data_format_, &dims));
403 std::vector<int32> strides_; member in class:tensorflow::Conv2DCustomBackpropFilterOp
452 OP_REQUIRES_OK(context, context->GetAttr("strides", &strides_));
453 int stride_n = GetTensorDim(strides_, data_format_, 'N')
531 std::vector<int32> strides_; member in class:tensorflow::Conv2DSlowBackpropFilterOp
[all...] |
| conv_grad_input_ops.cc | 277 OP_REQUIRES_OK(context, context->GetAttr("strides", &strides_));
278 OP_REQUIRES(context, strides_.size() == 4,
282 context, (strides_[0] == 1 && strides_[3] == 1),
285 OP_REQUIRES(context, strides_[1] > 0 && strides_[2] > 0,
331 strides_, padding_, data_format_, &dims));
537 std::vector<int32> strides_; member in class:tensorflow::Conv2DCustomBackpropInputOp
586 OP_REQUIRES_OK(context, context->GetAttr("strides", &strides_));
587 OP_REQUIRES(context, strides_.size() == 4
668 std::vector<int32> strides_; member in class:tensorflow::Conv2DSlowBackpropInputOp
[all...] |
| conv_ops_fused_image_transform.cc | 632 OP_REQUIRES_OK(context, context->GetAttr("strides", &strides_));
633 OP_REQUIRES(context, strides_.size() == 4,
636 const int64 stride_n = GetTensorDim(strides_, FORMAT_NHWC, 'N');
637 const int64 stride_c = GetTensorDim(strides_, FORMAT_NHWC, 'C');
866 std::vector<int32> strides_; member in class:tensorflow::FusedResizeConv2DUsingGemmOp
[all...] |
| depthwise_conv_grad_op.cc | 541 OP_REQUIRES_OK(context, context->GetAttr("strides", &strides_));
542 OP_REQUIRES(context, strides_.size() == 4,
551 stride_ = GetTensorDim(strides_, data_format_, 'H');
552 const int64 stride_w = GetTensorDim(strides_, data_format_, 'W');
553 const int64 stride_n = GetTensorDim(strides_, data_format_, 'N');
554 const int64 stride_c = GetTensorDim(strides_, data_format_, 'C');
653 std::vector<int32> strides_; member in class:tensorflow::DepthwiseConv2dNativeBackpropInputOp
1136 std::vector<int32> strides_; member in class:tensorflow::DepthwiseConv2dNativeBackpropFilterOp
[all...] |
| mkl_conv_ops.cc | 422 OP_REQUIRES_OK(context, context->GetAttr("strides", &strides_));
427 OP_REQUIRES(context, strides_.size() == 4,
431 const int64 stride_n = GetTensorDim(strides_, data_format_, 'N');
432 const int64 stride_c = GetTensorDim(strides_, data_format_, 'C');
522 const int stride_rows = GetTensorDim(strides_, data_format_, 'H');
523 const int stride_cols = GetTensorDim(strides_, data_format_, 'W');
834 std::vector<int32> strides_; member in class:tensorflow::MklConvOp
1282 std::vector<int32> strides_; member in class:tensorflow::MklConvOp
[all...] |
| /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/libaom/libaom/test/ |
| noise_model_test.cc | 534 strides_[c] = kWidth;
545 kWidth, kHeight, strides_, chroma_sub_,
564 int strides_[3]; member in class:NoiseModelUpdateTest
[all...] |
| /external/tensorflow/tensorflow/lite/delegates/nnapi/ |
| nnapi_delegate_test.cc | 1714 int strides_; member in class:tflite::__anon45929::StridedSliceOpModel
[all...] |
| /external/tensorflow/tensorflow/stream_executor/ |
| dnn.h | 689 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_; member in class:stream_executor::dnn::PoolingDescriptor
[all...] |
| /external/tensorflow/tensorflow/core/util/ |
| mkl_util.h | 160 if (strides_) delete[] strides_;
186 strides_ = new size_t[dimension];
190 strides_[ii] = strides[ii];
247 const size_t* GetStrides() const { return strides_; }
304 // strides_
315 // sizeof(strides_)
356 strides_ = new size_t[dimension_];
361 strides_[i] = reinterpret_cast<const size_t*>(
386 strides_[i]
408 size_t* strides_ = nullptr; \/\/ Required by MKL for conversions member in class:tensorflow::MklShape
[all...] |
