Searched defs:old_shape (Results 1 – 11 of 11) sorted by relevance
80 static Shape RowMajorShape(const Shape& old_shape) { in RowMajorShape()88 static Shape ColMajorShape(const Shape& old_shape) { in ColMajorShape()
248 const Shape& new_shape) { in TryResolvePaddedShapesForTensorCore()411 const Shape& new_shape) { in TryResolvePaddedShapesForIntegerConvolution()
42 TensorShape old_shape(shape); in compute_valid_region() local
127 const Shape& old_shape = conditional->operand(branch + 1)->shape(); in TryRemoveUnusedConditionalOperands() local245 const Shape old_shape = conditional_op->shape(); in RemoveUnusedTupleElements() local
198 Shape old_shape = hlo->shape(); in Run() local
1737 auto old_shape = consumer->mutable_operand(0)->shape(); in Propagate() local2007 auto old_shape = old_instr->shape(); in SelectValidPortion() local
690 const Shape& old_shape = old_root->shape(); in MoveInstructionIn() local
84 TensorShape old_shape = library->get_image_shape(old_image_name); in compute_target() local
1091 def _pad_left_to(n, old_shape): argument1141 def new_shape(_, old_shape): argument
395 inline std::vector<int64_t> GetPaddedShape(ArrayRef<int64_t> old_shape, in GetPaddedShape()1543 auto old_shape = shape.getType().getShape(); in matchAndRewrite() local
5482 xla::Shape old_shape = xla::TypeToShape(t); in GetTPUInfeedLayoutFromAPI() local