Searched refs:MakeShapeWithDescendingLayout (Results 1 – 10 of 10) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | multioutput_fusion_test.cc | 112 Literal arg1(ShapeUtil::MakeShapeWithDescendingLayout(F32, {size, size})); in RunTest2D()
115 Literal expect(ShapeUtil::MakeShapeWithDescendingLayout(F32, {size, size})); in RunTest2D()
128 ShapeUtil::MakeShapeWithDescendingLayout(F32, {size}); in RunTest1D()
130 ShapeUtil::MakeShapeWithDescendingLayout(F64, {size}); in RunTest1D()
150 ShapeUtil::MakeShapeWithDescendingLayout(F32, {size, 1}), add)); in RunTest1D()
155 ShapeUtil::MakeShapeWithDescendingLayout(F32, {1}), sub, reshape, in RunTest1D()
176 Literal input0(ShapeUtil::MakeShapeWithDescendingLayout(F32, {size})); in RunTest1D()
178 Literal input1(ShapeUtil::MakeShapeWithDescendingLayout(F64, {size})); in RunTest1D()
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | shape_util.cc | 320 /* static */ Shape ShapeUtil::MakeShapeWithDescendingLayout( in MakeShapeWithDescendingLayout() function in xla::ShapeUtil
334 Shape new_shape = MakeShapeWithDescendingLayout(shape.element_type(), dims); in MakeShapeWithDescendingLayoutAndSamePhysicalLayout()
1384 Shape input_shape_dim0_major = MakeShapeWithDescendingLayout( in ReshapeIsBitcast()
1386 Shape output_shape_dim0_major = MakeShapeWithDescendingLayout( in ReshapeIsBitcast()
1683 return ShapeUtil::MakeShapeWithDescendingLayout(shape.element_type(), in MergeDimensions()
|
| D | literal_util.h | 521 Literal literal(ShapeUtil::MakeShapeWithDescendingLayout( in CreateFullWithDescendingLayout()
|
| D | shape_util.h | 472 static Shape MakeShapeWithDescendingLayout(
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | cudnn_fused_conv_rewriter.cc | 227 ShapeUtil::MakeShapeWithDescendingLayout(conv_output_type, in TryRewriteToCudnnForwardRelu()
|
| D | ir_emitter_unnested.cc | 4105 Shape output_reduced_shape = ShapeUtil::MakeShapeWithDescendingLayout( in EmitTileElementForCopy()
4719 ShapeUtil::MakeShapeWithDescendingLayout( in EmitTilingKernel()
4871 Shape reduced_shape = ShapeUtil::MakeShapeWithDescendingLayout( in EmitHlo021Tile()
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | dot_op_emitter.cc | 1061 return ShapeUtil::MakeShapeWithDescendingLayout(shape.element_type(), in DropFirstDim()
1074 return ShapeUtil::MakeShapeWithDescendingLayout(shape.element_type(), in CollapseFirstNDims()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | algebraic_simplifier.cc | 5490 Shape new_input_shape = ShapeUtil::MakeShapeWithDescendingLayout( in SimplifyConvToDot()
5496 Shape new_filter_shape = ShapeUtil::MakeShapeWithDescendingLayout( in SimplifyConvToDot()
5499 Shape dot_output_shape = ShapeUtil::MakeShapeWithDescendingLayout( in SimplifyConvToDot()
|
| D | elemental_ir_emitter.cc | 2361 ShapeUtil::MakeShapeWithDescendingLayout( in MakeElementGenerator()
|
| D | algebraic_simplifier_test.cc | 2736 0, ShapeUtil::MakeShapeWithDescendingLayout(F32, {2, 2, 2}), in TEST_F()
|