Searched refs:MakeShapeWithDescendingLayout (Results 1 – 14 of 14) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/tests/ |
| H A D | multioutput_fusion_test.cc | 110 Literal arg1(ShapeUtil::MakeShapeWithDescendingLayout(F32, {size, size})); in RunTest2D()
113 Literal expect(ShapeUtil::MakeShapeWithDescendingLayout(F32, {size, size})); in RunTest2D()
126 ShapeUtil::MakeShapeWithDescendingLayout(F32, {size}); in RunTest1D()
128 ShapeUtil::MakeShapeWithDescendingLayout(F64, {size}); in RunTest1D()
148 ShapeUtil::MakeShapeWithDescendingLayout(F32, {size, 1}), add)); in RunTest1D()
153 ShapeUtil::MakeShapeWithDescendingLayout(F32, {1}), sub, reshape, in RunTest1D()
174 Literal input0(ShapeUtil::MakeShapeWithDescendingLayout(F32, {size})); in RunTest1D()
176 Literal input1(ShapeUtil::MakeShapeWithDescendingLayout(F64, {size})); in RunTest1D()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/ |
| H A D | bitcast_decomposer_test.cc | 63 ShapeUtil::MakeShapeWithDescendingLayout(F32, dims)); in AllPermutationsOfShapes()
97 ShapeUtil::MakeShapeWithDescendingLayout(F32, {4, 10, 100}),
98 ShapeUtil::MakeShapeWithDescendingLayout(F32, {4, 1, 10, 100}),
|
| H A D | reshape_decomposer.cc | 58 auto s0_normalized = ShapeUtil::MakeShapeWithDescendingLayout( in HandleReshape()
61 auto s_normalized = ShapeUtil::MakeShapeWithDescendingLayout( in HandleReshape()
|
| H A D | bitcast_decomposer.cc | 198 Shape reshape_shape = ShapeUtil::MakeShapeWithDescendingLayout( in Run()
|
| H A D | algebraic_simplifier.cc | 6637 Shape new_input_shape = ShapeUtil::MakeShapeWithDescendingLayout( in SimplifyConvToDot()
6643 Shape new_filter_shape = ShapeUtil::MakeShapeWithDescendingLayout( in SimplifyConvToDot()
6646 Shape dot_output_shape = ShapeUtil::MakeShapeWithDescendingLayout( in SimplifyConvToDot()
|
| H A D | elemental_ir_emitter.cc | 2556 ShapeUtil::MakeShapeWithDescendingLayout( in MakeElementGenerator()
|
| H A D | algebraic_simplifier_test.cc | 3144 0, ShapeUtil::MakeShapeWithDescendingLayout(F32, {2, 2, 2}), in TEST_F()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/ |
| H A D | shape_util.cc | 351 /* static */ Shape ShapeUtil::MakeShapeWithDescendingLayout( in MakeShapeWithDescendingLayout() function in xla::ShapeUtil
369 Shape new_shape = MakeShapeWithDescendingLayout(shape.element_type(), dims); in MakeShapeWithDescendingLayoutAndSamePhysicalLayout()
1466 Shape input_shape_dim0_major = MakeShapeWithDescendingLayout( in ReshapeIsBitcast()
1468 Shape output_shape_dim0_major = MakeShapeWithDescendingLayout( in ReshapeIsBitcast()
1688 return ShapeUtil::MakeShapeWithDescendingLayout(shape.element_type(), in MergeDimensions()
|
| H A D | shape_util.h | 370 static Shape MakeShapeWithDescendingLayout(
|
| H A D | literal_util.h | 513 Literal literal(ShapeUtil::MakeShapeWithDescendingLayout( in CreateFullWithDescendingLayout()
|
| H A D | xla.proto | 421 // ShapeUtil::MakeShapeWithDescendingLayout is an identity on all
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| H A D | dot_op_emitter.cc | 1258 return ShapeUtil::MakeShapeWithDescendingLayout(shape.element_type(), in DropFirstDim()
1271 return ShapeUtil::MakeShapeWithDescendingLayout(shape.element_type(), in CollapseFirstNDims()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/python/ |
| H A D | py_client.cc | 622 ShapeUtil::MakeShapeWithDescendingLayout(U64, {}); in EmitPythonCallbackFromDescriptor()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| H A D | ir_emitter_unnested.cc | 4126 ShapeUtil::MakeShapeWithDescendingLayout( in EmitTilingKernel()
4239 Shape reduced_shape = ShapeUtil::MakeShapeWithDescendingLayout( in EmitHlo021Tile()
|