Searched refs:MakeShapeWithSparseLayout (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | layout_util_test.cc | 37 Shape MakeShapeWithSparseLayout(PrimitiveType element_type, in MakeShapeWithSparseLayout() function in xla::__anon976bbfea0111::LayoutUtilTest
96 Shape src = MakeShapeWithSparseLayout(F32, {2, 3}, 2); in TEST_F()
139 {MakeShapeWithSparseLayout(F32, {2, 3}, 4), in TEST_F()
140 MakeShapeWithSparseLayout(F32, {42, 123}, 4), in TEST_F()
143 MakeShapeWithSparseLayout(F32, {1, 2, 3}, 6)})}); in TEST_F()
164 Shape src = MakeShapeWithSparseLayout(F32, {123, 42, 7}, 6); in TEST_F()
181 Shape dst = MakeShapeWithSparseLayout(F32, {2, 3}, 4); in TEST_F()
|
| D | shape_util.h | 405 static Shape MakeShapeWithSparseLayout(PrimitiveType element_type,
|
| D | literal_util.h | 420 Literal literal(ShapeUtil::MakeShapeWithSparseLayout( in CreateSparse()
|
| D | shape_util.cc | 205 /* static */ Shape ShapeUtil::MakeShapeWithSparseLayout( in MakeShapeWithSparseLayout() function in xla::ShapeUtil
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | cpu_hlo_support_checker_test.cc | 53 const Shape sparse_shape = ShapeUtil::MakeShapeWithSparseLayout(F32, {10}, 2); in TEST_F()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | gpu_hlo_support_checker_test.cc | 53 const Shape sparse_shape = ShapeUtil::MakeShapeWithSparseLayout(F32, {10}, 2); in TEST_F()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | pattern_matcher_test.cc | 107 auto array_shape = ShapeUtil::MakeShapeWithSparseLayout(F32, {2, 3, 4}, 10); in TEST()
|