Searched refs:ShapeSizeInBytes (Results 1 – 6 of 6) sorted by relevance
225 ShapeSizeInBytes(output_shape) < ShapeSizeInBytes(operand.base_shape())) { in ParititonTrivialIndexedOperandDimension()637 ShapeSizeInBytes(updates.base_shape()) < in HandleScatter()638 ShapeSizeInBytes(scatter->shape())) { in HandleScatter()
270 if (ShapeSizeInBytes(lhs.base_shape()) < ShapeSizeInBytes(rhs.base_shape())) { in PartitionConvolutionWithSpatialDimensionHaloExchangeOnRHS()579 if (ShapeSizeInBytes(lhs.base_shape()) < ShapeSizeInBytes(rhs.base_shape())) { in PartitionConvolutionWithSpatialDimensionHaloExchangeOnLHS()
1474 ShapeSizeInBytes(rhs.base_shape()), in PartitionBaseCase()1475 ShapeSizeInBytes(lhs.base_shape()), in PartitionBaseCase()1476 ShapeSizeInBytes(output_base_shape), in PartitionBaseCase()1501 if (ShapeSizeInBytes(lhs.base_shape()) < in PartitionBaseCase()1502 ShapeSizeInBytes(rhs.base_shape())) { in PartitionBaseCase()2404 ShapeSizeInBytes(other_partitioned.base_shape()), in LhsIsBestMatchForNonContractingPartitioning()2405 ShapeSizeInBytes(matching_partitioned.base_shape()) / in LhsIsBestMatchForNonContractingPartitioning()2407 ShapeSizeInBytes( in LhsIsBestMatchForNonContractingPartitioning()
93 int64 ShapeSizeInBytes(const Shape& shape);
90 max_value = std::max<int64>(max_value, ShapeSizeInBytes(inst->shape())); in RegisterLogEntry()154 return ShapeSizeInBytes(inst0->shape()) > in ReportMemoryUsage()155 ShapeSizeInBytes(inst1->shape()); in ReportMemoryUsage()161 ShapeSizeInBytes((*insts)[i]->shape())), in ReportMemoryUsage()
161 int64 ShapeSizeInBytes(const Shape& shape) { in ShapeSizeInBytes() function