Home
last modified time | relevance | path

Searched refs:MakeShapeWithDescendingLayoutAndSamePhysicalLayout (Results 1 – 7 of 7) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dlayout_normalization.cc105 ShapeUtil::MakeShapeWithDescendingLayoutAndSamePhysicalLayout( in HandleConcatenate()
120 ShapeUtil::MakeShapeWithDescendingLayoutAndSamePhysicalLayout(s); in HandleConcatenate()
122 ShapeUtil::MakeShapeWithDescendingLayoutAndSamePhysicalLayout( in HandleConcatenate()
366 ShapeUtil::MakeShapeWithDescendingLayoutAndSamePhysicalLayout(s); in HandleReverse()
398 ShapeUtil::MakeShapeWithDescendingLayoutAndSamePhysicalLayout( in HandlePad()
408 ShapeUtil::MakeShapeWithDescendingLayoutAndSamePhysicalLayout(s); in HandlePad()
518 ShapeUtil::MakeShapeWithDescendingLayoutAndSamePhysicalLayout(s)); in Normalize()
H A Dbitcast_decomposer.cc164 ShapeUtil::MakeShapeWithDescendingLayoutAndSamePhysicalLayout( in Run()
H A Dhlo_rematerialization_test.cc569 ShapeUtil::MakeShapeWithDescendingLayoutAndSamePhysicalLayout(shape); in ShapeSizePadMinorTo64()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/
H A Dshape_util.h378 static Shape MakeShapeWithDescendingLayoutAndSamePhysicalLayout(
H A Dshape_util.cc359 ShapeUtil::MakeShapeWithDescendingLayoutAndSamePhysicalLayout( in MakeShapeWithDescendingLayoutAndSamePhysicalLayout() function in xla::ShapeUtil
1712 ShapeUtil::MakeShapeWithDescendingLayoutAndSamePhysicalLayout(a); in FindTranspose021()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/
H A Dmhlo_to_lhlo_with_xla.cc1462 xla::ShapeUtil::MakeShapeWithDescendingLayoutAndSamePhysicalLayout( in GetOrCreateArrayView()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/cpu/
H A Dir_emitter.cc599 ShapeUtil::MakeShapeWithDescendingLayoutAndSamePhysicalLayout(keys_shape); in HandleSort()