Searched refs:ByteSizeOfElements (Results 1 – 8 of 8) sorted by relevance
151 ShapeUtil::ByteSizeOfElements(consumer->operand(0)->shape()) < in ShouldFuse()157 ShapeUtil::ByteSizeOfElements(consumer->operand(1)->shape()) < in ShouldFuse()
35 ShapeSizeFunction shape_size_function = ShapeUtil::ByteSizeOfElements)
58 ShapeSizeFunction shape_size_function = ShapeUtil::ByteSizeOfElements)
221 int64_t bytes = ShapeUtil::ByteSizeOfElements(instr->shape()); in MergeDots()223 bytes += ShapeUtil::ByteSizeOfElements(operand->shape()); in MergeDots()
45 ShapeUtil::ByteSizeOfElements); in CreateNewVerifiedModuleWithFTZ()
49 /*shape_size_function=*/ShapeUtil::ByteSizeOfElements); in CompileHloConvAndGetMlir()
143 static int64_t ByteSizeOfElements(const Shape& shape);
829 return ByteSizeOfElements(shape); in ByteSizeOf()848 /* static */ int64_t ShapeUtil::ByteSizeOfElements(const Shape& shape) { in ByteSizeOfElements() function in xla::ShapeUtil