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 | 225 /* static */ Shape ShapeUtil::MakeShapeWithDescendingLayout( in MakeShapeWithDescendingLayout() function in xla::ShapeUtil
239 Shape new_shape = MakeShapeWithDescendingLayout(shape.element_type(), dims); in MakeShapeWithDescendingLayoutAndSamePhysicalLayout()
1253 Shape input_shape_dim0_major = MakeShapeWithDescendingLayout( in ReshapeIsBitcast()
1255 Shape output_shape_dim0_major = MakeShapeWithDescendingLayout( in ReshapeIsBitcast()
1509 return ShapeUtil::MakeShapeWithDescendingLayout(shape.element_type(), in MergeDimensions()
|
| D | literal_util.h | 521 Literal literal(ShapeUtil::MakeShapeWithDescendingLayout( in CreateFullWithDescendingLayout()
|
| D | shape_util.h | 423 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 | 2001 Shape output_reduced_shape = ShapeUtil::MakeShapeWithDescendingLayout( in EmitTileElementForCopy()
2432 ShapeUtil::MakeShapeWithDescendingLayout( in EmitTilingKernel()
2577 Shape reduced_shape = ShapeUtil::MakeShapeWithDescendingLayout( in EmitHlo021Tile()
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | dot_op_emitter.cc | 917 return ShapeUtil::MakeShapeWithDescendingLayout(shape.element_type(), in DropFirstDim()
930 return ShapeUtil::MakeShapeWithDescendingLayout(shape.element_type(), in CollapseFirstNDims()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | algebraic_simplifier.cc | 4322 Shape new_input_shape = ShapeUtil::MakeShapeWithDescendingLayout( in SimplifyConvToDot()
4328 Shape new_filter_shape = ShapeUtil::MakeShapeWithDescendingLayout( in SimplifyConvToDot()
4331 Shape dot_output_shape = ShapeUtil::MakeShapeWithDescendingLayout( in SimplifyConvToDot()
|
| D | elemental_ir_emitter.cc | 2262 ShapeUtil::MakeShapeWithDescendingLayout( in MakeElementGenerator()
|
| D | algebraic_simplifier_test.cc | 2506 0, ShapeUtil::MakeShapeWithDescendingLayout(F32, {2, 2, 2}), in TEST_F()
|