Searched refs:ByteSizeOfElements (Results 1 – 9 of 9) sorted by relevance
87 /*shape_size_function=*/ShapeUtil::ByteSizeOfElements); in CompileAndVerifyIr()138 /*shape_size_function=*/ShapeUtil::ByteSizeOfElements); in CompileAndVerifyErrors()
158 ShapeUtil::ByteSizeOfElements(consumer->operand(0)->shape()) < in ShouldFuse()164 ShapeUtil::ByteSizeOfElements(consumer->operand(1)->shape()) < in ShouldFuse()
44 ShapeSizeFunction shape_size_function = ShapeUtil::ByteSizeOfElements)
48 ShapeUtil::ByteSizeOfElements); in GetParsedModule()
46 ShapeUtil::ByteSizeOfElements); in CreateNewVerifiedModuleWithFTZ()
43 /*shape_size_function=*/ShapeUtil::ByteSizeOfElements); in CompileHloConvAndGetMlir()
629 int64 byte_size = ByteSizeOfElements(shape); in ByteSizeOf()649 /* static */ int64 ShapeUtil::ByteSizeOfElements(const Shape& shape) { in ByteSizeOfElements() function in xla::ShapeUtil
232 static int64 ByteSizeOfElements(const Shape& shape);
1563 int64 num_bytes = ShapeUtil::ByteSizeOfElements(init_value->shape()); in BuildInitializerThunk()