Searched refs:ByteSizeOfElements (Results 1 – 4 of 4) sorted by relevance
225 input_size += ShapeUtil::ByteSizeOfElements(subshape); in TryHoistingInvariantInstructionsFromWhileBody()233 output_size += ShapeUtil::ByteSizeOfElements(subshape); in TryHoistingInvariantInstructionsFromWhileBody()
576 int64 byte_size = ByteSizeOfElements(shape); in ByteSizeOf()599 /* static */ int64 ShapeUtil::ByteSizeOfElements(const Shape& shape) { in ByteSizeOfElements() function in xla::ShapeUtil
227 static int64 ByteSizeOfElements(const Shape& shape);
1946 int64 num_bytes = ShapeUtil::ByteSizeOfElements(init_value->shape()); in BuildInitializerThunk()