Searched refs:ByteSizeOfElements (Results 1 – 8 of 8) sorted by relevance
157 ShapeUtil::ByteSizeOfElements(consumer->operand(0)->shape()) < in ShouldFuse()163 ShapeUtil::ByteSizeOfElements(consumer->operand(1)->shape()) < in ShouldFuse()
44 ShapeSizeFunction shape_size_function = ShapeUtil::ByteSizeOfElements)
35 ShapeSizeFunction shape_size_function = ShapeUtil::ByteSizeOfElements)
48 ShapeUtil::ByteSizeOfElements); in GetParsedModule()
46 ShapeUtil::ByteSizeOfElements); in CreateNewVerifiedModuleWithFTZ()
45 /*shape_size_function=*/ShapeUtil::ByteSizeOfElements); in CompileHloConvAndGetMlir()
233 static int64 ByteSizeOfElements(const Shape& shape);
748 return ByteSizeOfElements(shape); in ByteSizeOf()767 /* static */ int64 ShapeUtil::ByteSizeOfElements(const Shape& shape) { in ByteSizeOfElements() function in xla::ShapeUtil