Home
last modified time | relevance | path

Searched defs:old_shape (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dcpu_layout_assignment.cc80 static Shape RowMajorShape(const Shape& old_shape) { in RowMajorShape()
88 static Shape ColMajorShape(const Shape& old_shape) { in ColMajorShape()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dcudnn_pad_for_convolutions.cc245 const Shape& new_shape) { in TryResolvePaddedShapesForTensorCore()
354 const Shape& new_shape) { in TryResolvePaddedShapesForIntegerConvolution()
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_element_type_converter.cc195 Shape old_shape = hlo->shape(); in Run() local
Dconditional_simplifier.cc263 const Shape& old_shape = conditional->operand(branch + 1)->shape(); in TryRemoveUnusedConditionalOperands() local
381 const Shape old_shape = conditional_op->shape(); in RemoveUnusedTupleElements() local
Dspace_to_batch_converter.cc1419 auto old_shape = consumer->mutable_operand(0)->shape(); in Propagate() local
1654 auto old_shape = old_instr->shape(); in SelectValidPortion() local
Dconditional_code_motion.cc688 const Shape& old_shape = old_root->shape(); in MoveInstructionIn() local
/external/tensorflow/tensorflow/python/ops/numpy_ops/
Dnp_array_ops.py1085 def _pad_left_to(n, old_shape): argument
1135 def new_shape(_, old_shape): argument
/external/tensorflow/tensorflow/compiler/mlir/lite/ir/
Dtfl_ops.cc342 inline std::vector<int64_t> GetPaddedShape(ArrayRef<int64_t> old_shape, in GetPaddedShape()
1290 auto old_shape = shape.getType().getShape(); in matchAndRewrite() local