Searched defs:old_shape (Results 1 – 12 of 12) sorted by relevance
81 static Shape RowMajorShape(const Shape& old_shape) { in RowMajorShape()89 static Shape ColMajorShape(const Shape& old_shape) { in ColMajorShape()
258 const Shape& new_shape) { in TryResolvePaddedShapesForTensorCore()428 const Shape& new_shape) { in TryResolvePaddedShapesForIntegerConvolution()
129 const Shape& old_shape = conditional->operand(branch + 1)->shape(); in TryRemoveUnusedConditionalOperands() local247 const Shape old_shape = conditional_op->shape(); in RemoveUnusedTupleElements() local
204 Shape old_shape = hlo->shape(); in Run() local
2022 auto old_shape = consumer->mutable_operand(0)->shape(); in Propagate() local2292 auto old_shape = old_instr->shape(); in SelectValidPortion() local
779 const Shape& old_shape = old_root->shape(); in MoveInstructionIn() local
42 TensorShape old_shape(shape); in compute_valid_region() local
48 xla::Shape old_shape = xla::TypeToShape(t); in GetTPUInfeedLayoutFromAPI() local
1087 def _pad_left_to(n, old_shape): argument1137 def new_shape(_, old_shape): argument
294 auto old_shape = self_.sym_sizes(); in roll_batch_rule() local
437 const auto old_shape = shape.cast<DenseIntElementsAttr>(); in ReshapeValueDroppingLastDim() local
484 inline std::vector<int64_t> GetPaddedShape(ArrayRef<int64_t> old_shape, in GetPaddedShape()1688 auto old_shape = shape.getType().getShape(); in matchAndRewrite() local