Searched defs:input_spatial_dims (Results 1 – 6 of 6) sorted by relevance
104 auto input_spatial_dims = conv_dnums.input_spatial_dimensions(); in MatchBackwardFilter() local346 const auto& input_spatial_dims = dnums.input_spatial_dimensions(); in MatchBackwardInput() local
48 llvm::ArrayRef<int64_t> input_spatial_dims; member
90 llvm::SmallVector<int64_t, 4> input_spatial_dims( in ConvertConvDimensionNumbers() local
1735 std::vector<int64_t> input_spatial_dims(num_spatial_dims); in InferConvolveShape() local
1559 std::vector<int64_t> input_spatial_dims; in InferXlaConvOutputShape() local
646 gtl::InlinedVector<DimensionHandle, 2> input_spatial_dims(2); in Conv2DShapeImpl() local