Searched refs:MakeShapeWithDescendingLayout (Results 1 – 12 of 12) 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/service/llvm_ir/ |
| D | kernel_tiling.cc | 52 return ShapeUtil::MakeShapeWithDescendingLayout(shape.element_type(), in MergeDimensions()
153 Shape output_shape = ShapeUtil::MakeShapeWithDescendingLayout( in GetUnnormalizedIndex()
167 ShapeUtil::MakeShapeWithDescendingLayout( in EmitBlockIndex()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | index_ops_cpu.cc | 106 xla::Shape xla_shape = xla::ShapeUtil::MakeShapeWithDescendingLayout( in Compile()
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | shape_util.cc | 198 /* static */ Shape ShapeUtil::MakeShapeWithDescendingLayout( in MakeShapeWithDescendingLayout() function in xla::ShapeUtil
222 return MakeShapeWithDescendingLayout(shape.element_type(), dims); in MakeShapeWithDescendingLayoutAndSamePhysicalLayout()
1226 Shape input_shape_dim0_major = MakeShapeWithDescendingLayout( in ReshapeIsBitcast()
1228 Shape output_shape_dim0_major = MakeShapeWithDescendingLayout( in ReshapeIsBitcast()
|
| D | shape_util.h | 410 static Shape MakeShapeWithDescendingLayout(
|
| D | literal_util.h | 554 Literal literal(ShapeUtil::MakeShapeWithDescendingLayout( in CreateFullWithDescendingLayout()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | cudnn_fused_conv_rewriter.cc | 213 ShapeUtil::MakeShapeWithDescendingLayout(element_type, in TryRewriteToCudnnForwardRelu()
|
| D | ir_emitter_unnested.cc | 2287 param_reduced_shapes->push_back(ShapeUtil::MakeShapeWithDescendingLayout( in ConstructInputReducedShapeAndCastInputIrArrayToShape()
2467 Shape output_reduced_shape = ShapeUtil::MakeShapeWithDescendingLayout( in EmitTileElementForCopy()
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | dot_op_emitter.cc | 865 return ShapeUtil::MakeShapeWithDescendingLayout(shape.element_type(), in DropFirstDim()
878 return ShapeUtil::MakeShapeWithDescendingLayout(shape.element_type(), in CollapseFirstNDims()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | algebraic_simplifier.cc | 3617 const Shape new_input_shape = ShapeUtil::MakeShapeWithDescendingLayout( in SimplifyConvToDot()
3622 const Shape new_filter_shape = ShapeUtil::MakeShapeWithDescendingLayout( in SimplifyConvToDot()
3624 const Shape dot_output_shape = ShapeUtil::MakeShapeWithDescendingLayout( in SimplifyConvToDot()
|
| D | elemental_ir_emitter.cc | 2317 ShapeUtil::MakeShapeWithDescendingLayout( in MakeElementGenerator()
|
| D | algebraic_simplifier_test.cc | 2191 0, ShapeUtil::MakeShapeWithDescendingLayout(F32, {2, 2, 2}), in TEST_F()
|