Searched defs:flattened_size (Results 1 – 2 of 2) sorted by relevance
159 int64_t flattened_size = xla::Product(iota_shape.dimensions()); in CompileWhereWithSort() local208 int64_t flattened_size = xla::Product(input_shape.dimensions()); in CompileWhereWithPrefixSum() local
841 int32_t flattened_size = 1; in ConvertSoftmaxOperator() local883 int32_t flattened_size = 1; in ConvertLogSoftmaxOperator() local