Searched defs:partitions_1d (Results 1 – 1 of 1) sorted by relevance
63 xla::XlaOp partitions_1d, const xla::Shape& data_1d_shape, in DynamicPartition1D()158 auto partitions_1d = xla::Reshape(partitions, {input_count}); in Compile() local