Home
last modified time | relevance | path

Searched refs:IsScalarWithElementType (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_query.cc27 ShapeUtil::IsScalarWithElementType(instruction->shape(), F32)) { in IsConstantR0F32()
Dhlo_instruction_test.cc139 EXPECT_TRUE(ShapeUtil::IsScalarWithElementType(parameter->shape(), F32)); in TEST_F()
140 EXPECT_FALSE(ShapeUtil::IsScalarWithElementType(parameter->shape(), S32)); in TEST_F()
/external/tensorflow/tensorflow/compiler/xla/
Dshape_util.h319 static bool IsScalarWithElementType(const Shape& shape,
Dshape_util.cc418 /* static */ bool ShapeUtil::IsScalarWithElementType( in IsScalarWithElementType() function in xla::ShapeUtil