Searched refs:stride_dims (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/python/keras/utils/ |
| D | conv_utils.py | 285 stride_dims = len(strides)
286 if kernel_dims != in_dims or stride_dims != in_dims:
289 (stride_dims, in_dims, kernel_dims))
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | randomized_tests.cc | 331 std::vector<int64> stride_dims; member
654 d.stride_dims.resize(num_spatial_dims); in ChooseWindowedSpatialDims()
663 d.stride_dims[i] = in ChooseWindowedSpatialDims()
667 d.stride_dims[i], d.padding, &d.output_dims[i], in ChooseWindowedSpatialDims()
1135 std::vector<int64> input_dims, kernel_dims, stride_dims; in TEST_F() local
1140 stride_dims.push_back(random_int(generator())); in TEST_F()
1152 .Attr("strides", ImageDims(FORMAT_NHWC, 1, 1, stride_dims)) in TEST_F()
1173 .Attr("strides", ImageDims(FORMAT_NHWC, 1, 1, d.stride_dims)) in TEST_F()
1193 .Attr("strides", ImageDims(FORMAT_NHWC, 1, 1, d.stride_dims)) in TEST_F()
1614 .Attr("strides", ImageDims(FORMAT_NHWC, 1, 1, d.stride_dims)) in TEST_F()
[all …]
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_nodes.cc | 2325 nvinfer1::Dims begin_dims, size_dims, stride_dims; in ConvertStridedSliceHelper() local
2330 TF_RETURN_IF_ERROR(TensorShapeArrayToTrtDims(stride, &stride_dims, in ConvertStridedSliceHelper()
2336 stride_dims); in ConvertStridedSliceHelper()
|