Home
last modified time | relevance | path

Searched refs:IsEmptyTuple (Results 1 – 10 of 10) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Doutfeed_thunk.cc42 if (ShapeUtil::IsEmptyTuple(hlo_instruction()->operand(0)->shape())) { in ExecuteOnStream()
/external/tensorflow/tensorflow/compiler/xla/
Dshape_util_test.cc305 EXPECT_TRUE(ShapeUtil::IsEmptyTuple(ShapeUtil::MakeNil())); in TEST()
306 EXPECT_FALSE(ShapeUtil::IsEmptyTuple(ShapeUtil::MakeShape(F32, {1, 2, 3}))); in TEST()
307 EXPECT_FALSE(ShapeUtil::IsEmptyTuple(ShapeUtil::MakeShape(F32, {0, 1}))); in TEST()
308 EXPECT_FALSE(ShapeUtil::IsEmptyTuple( in TEST()
310 EXPECT_FALSE(ShapeUtil::IsEmptyTuple( in TEST()
Dshape_tree_test.cc55 EXPECT_TRUE(ShapeUtil::IsEmptyTuple(int_tree.shape())); in TEST_F()
58 EXPECT_TRUE(ShapeUtil::IsEmptyTuple(bool_tree.shape())); in TEST_F()
Dshape_util.h465 static bool IsEmptyTuple(const Shape& shape);
Dliteral_test.cc1670 EXPECT_FALSE(ShapeUtil::IsEmptyTuple(nested_tuple.shape())); in TEST_F()
1672 EXPECT_TRUE(ShapeUtil::IsEmptyTuple(nested_tuple.shape())); in TEST_F()
1723 EXPECT_TRUE(ShapeUtil::IsEmptyTuple(element.shape())); in TEST_F()
Dshape_util.cc334 /* static */ bool ShapeUtil::IsEmptyTuple(const Shape& shape) { in IsEmptyTuple() function in xla::ShapeUtil
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_sharding.cc205 return ShapeUtil::IsEmptyTuple(shape) ? 1 : ShapeUtil::GetLeafCount(shape); in RequiredLeaves()
225 if (ShapeUtil::IsEmptyTuple(shape)) { in AsShapeTree()
Dwhile_loop_simplifier.cc405 if (ShapeUtil::IsEmptyTuple(new_while_shape)) { in TryRemoveConstantParams()
Dhlo_parser.cc1399 if (!shape.IsTuple() && !ShapeUtil::IsEmptyTuple(shape)) { in ParseInstructionRhs()
/external/tensorflow/tensorflow/compiler/xla/client/
Dclient.cc317 if (!ShapeUtil::IsEmptyTuple(shape)) { in Execute()