Searched defs:expand_shape (Results 1 – 7 of 7) sorted by relevance
111 auto expand_shape = input.sym_sizes().vec(); in repeat_interleave_symint() local
109 def expand_shape(arr_shape, axis): function
323 Value expand_shape = b.create<ExpandShapeOp>( in TileAndReshapeInputTensors() local
161 DimVector expand_shape(tensor_.sizes().begin(), tensor_.sizes().end()); in make_dummy() local
185 SmallVector<int64_t, 4> expand_shape(pad_shape.begin(), pad_shape.end()); in GetPadOpForConv2DFilter() local
2398 auto expand_shape = DimVector(B.sizes().slice(0, n - 2)); in maybe_expand_lu() local2410 auto expand_shape = DimVector(B.sizes().slice(0, B.dim() - 2)); in maybe_expand_pivots() local
1399 auto expand_shape = in ComputeConstant() local