Searched refs:vect_size (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | cudnn_vectorize_convolutions.cc | 92 static XlaOp SplitAtDim(XlaOp instr, int64_t dim, int64_t vect_size) { in SplitAtDim() argument
97 CHECK_EQ(new_dims[dim] % vect_size, 0); in SplitAtDim()
98 new_dims[dim] /= vect_size; in SplitAtDim()
99 new_dims.insert(new_dims.begin() + dim + 1, vect_size); in SplitAtDim()
108 static Shape SplitShapeAtDim(Shape shape, int64_t dim, int64_t vect_size) { in SplitShapeAtDim() argument
111 CHECK_EQ(new_dims[dim] % vect_size, 0); in SplitShapeAtDim()
112 new_dims[dim] /= vect_size; in SplitShapeAtDim()
113 new_dims.insert(new_dims.begin() + dim + 1, vect_size); in SplitShapeAtDim()
141 int64_t vect_size) { in RevectorizeInstr() argument
146 CHECK_LE(size(vect_dim), vect_size); in RevectorizeInstr()
[all …]
|
| D | stream_executor_util.cc | 203 int64_t vect_size = input.dimensions(input.layout().minor_to_major(0)); in XlaConvShapesToStreamExecutorLayouts() local
204 if (vect_size == 4) { in XlaConvShapesToStreamExecutorLayouts()
206 } else if (vect_size == 32) { in XlaConvShapesToStreamExecutorLayouts()
213 ConvolutionDimensionNumbersToString(dnums), vect_size); in XlaConvShapesToStreamExecutorLayouts()
227 int64_t vect_size = filter.dimensions(filter.layout().minor_to_major(0)); in XlaConvShapesToStreamExecutorLayouts() local
228 if (vect_size == 4) { in XlaConvShapesToStreamExecutorLayouts()
230 } else if (vect_size == 32) { in XlaConvShapesToStreamExecutorLayouts()
237 ConvolutionDimensionNumbersToString(dnums), vect_size); in XlaConvShapesToStreamExecutorLayouts()
251 int64_t vect_size = output.dimensions(output.layout().minor_to_major(0)); in XlaConvShapesToStreamExecutorLayouts() local
252 if (vect_size == 4) { in XlaConvShapesToStreamExecutorLayouts()
[all …]
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | common_shape_fns.cc | 552 absl::optional<DimensionHandle> vect_size, in ShapeFromDimensions() argument
570 CHECK(vect_size.has_value()); // Crash ok. in ShapeFromDimensions()
572 filter_dim, *vect_size, /*evenly_divisible=*/true, in ShapeFromDimensions()
574 out_dims[GetTensorInnerFeatureDimIndex(rank, format)] = *vect_size; in ShapeFromDimensions()
744 absl::optional<DimensionHandle> vect_size; in Conv2DShapeImpl() local
746 vect_size.emplace(c->Dim(conv_input_shape, in Conv2DShapeImpl()
752 vect_size, c, &output_shape)); in Conv2DShapeImpl()
|