Searched refs:MakeShapeWithSparseLayout (Results 1 - 7 of 7) sorted by relevance
/external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
H A D | cpu_hlo_support_checker_test.cc | 53 const Shape sparse_shape = ShapeUtil::MakeShapeWithSparseLayout(F32, {10}, 2);
|
/external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
H A D | gpu_hlo_support_checker_test.cc | 53 const Shape sparse_shape = ShapeUtil::MakeShapeWithSparseLayout(F32, {10}, 2);
|
/external/tensorflow/tensorflow/compiler/xla/ |
H A D | layout_util_test.cc | 37 Shape MakeShapeWithSparseLayout(PrimitiveType element_type, function in class:xla::__anon25506::LayoutUtilTest 96 Shape src = MakeShapeWithSparseLayout(F32, {2, 3}, 2); 139 {MakeShapeWithSparseLayout(F32, {2, 3}, 4), 140 MakeShapeWithSparseLayout(F32, {42, 123}, 4), 143 MakeShapeWithSparseLayout(F32, {1, 2, 3}, 6)})}); 164 Shape src = MakeShapeWithSparseLayout(F32, {123, 42, 7}, 6); 181 Shape dst = MakeShapeWithSparseLayout(F32, {2, 3}, 4);
|
H A D | shape_util.h | 329 static Shape MakeShapeWithSparseLayout(
|
H A D | shape_util.cc | 205 /* static */ Shape ShapeUtil::MakeShapeWithSparseLayout( function in class:ShapeUtil 578 result = ShapeUtil::MakeShapeWithSparseLayout(primitive_type, dimensions,
|
H A D | shape_util_test.cc | 133 Shape expected = ShapeUtil::MakeShapeWithSparseLayout(F32, {123, 456}, 10);
|
H A D | literal_util.h | 988 auto literal = MakeUnique<Literal>(ShapeUtil::MakeShapeWithSparseLayout(
|
Completed in 253 milliseconds