HomeSort by relevance Sort by last modified time
    Searched refs:IsScalarWithElementType (Results 1 - 4 of 4) sorted by null

  /external/tensorflow/tensorflow/compiler/xla/service/
hlo_query.cc 27 ShapeUtil::IsScalarWithElementType(instruction->shape(), F32)) {
hlo_instruction_test.cc 139 EXPECT_TRUE(ShapeUtil::IsScalarWithElementType(parameter->shape(), F32));
140 EXPECT_FALSE(ShapeUtil::IsScalarWithElementType(parameter->shape(), S32));
    [all...]
  /external/tensorflow/tensorflow/compiler/xla/
shape_util.h 319 static bool IsScalarWithElementType(const Shape& shape,
    [all...]
shape_util.cc 418 /* static */ bool ShapeUtil::IsScalarWithElementType(
    [all...]

Completed in 98 milliseconds