Searched defs:data_1d (Results 1 – 1 of 1) sorted by relevance
62 DynamicPartition1D(XlaOpKernelContext* ctx, xla::XlaOp data_1d, in DynamicPartition1D()157 auto data_1d = xla::Reshape(data, {input_count}); in Compile() local