Searched defs:data_1d_shape (Results 1 – 1 of 1) sorted by relevance
63 xla::XlaOp partitions_1d, const xla::Shape& data_1d_shape, in DynamicPartition1D()159 xla::Shape data_1d_shape = in Compile() local