Searched refs:stride_dims (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/python/keras/utils/ |
| D | conv_utils.py | 272 stride_dims = len(strides)
273 if kernel_dims != in_dims or stride_dims != in_dims:
276 (stride_dims, in_dims, kernel_dims))
354 stride_dims = len(strides)
355 if kernel_dims != in_dims or stride_dims != in_dims:
358 (stride_dims, in_dims, kernel_dims))
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | randomized_tests.cc | 332 std::vector<int64> stride_dims; member
655 d.stride_dims.resize(num_spatial_dims); in ChooseWindowedSpatialDims()
664 d.stride_dims[i] = in ChooseWindowedSpatialDims()
668 d.stride_dims[i], d.padding, &d.output_dims[i], in ChooseWindowedSpatialDims()
1136 std::vector<int64> input_dims, kernel_dims, stride_dims; in TEST_F() local
1141 stride_dims.push_back(random_int(generator())); in TEST_F()
1153 .Attr("strides", ImageDims(FORMAT_NHWC, 1, 1, stride_dims)) in TEST_F()
1174 .Attr("strides", ImageDims(FORMAT_NHWC, 1, 1, d.stride_dims)) in TEST_F()
1194 .Attr("strides", ImageDims(FORMAT_NHWC, 1, 1, d.stride_dims)) in TEST_F()
1615 .Attr("strides", ImageDims(FORMAT_NHWC, 1, 1, d.stride_dims)) in TEST_F()
[all …]
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_nodes.cc | 2879 nvinfer1::Dims begin_dims, size_dims, stride_dims; in ConvertStridedSliceHelper() local
2884 TF_RETURN_IF_ERROR(TensorShapeArrayToTrtDims(stride, &stride_dims, in ConvertStridedSliceHelper()
2889 *input.tensor(), begin_dims, size_dims, stride_dims); in ConvertStridedSliceHelper()
|