Searched defs:flattened_size (Results 1 – 2 of 2) sorted by relevance
47 int64 flattened_size = xla::Product(iota_shape.dimensions()); in Compile() local
836 int32 flattened_size = 1; in ConvertSoftmaxOperator() local878 int32 flattened_size = 1; in ConvertLogSoftmaxOperator() local