Searched refs:leading_dim (Results 1 – 2 of 2) sorted by relevance
65 int64 leading_dim, DataType dtype, xla::XlaOp* list) { in CreateZerosList() argument67 list_shape.AddDim(leading_dim); in CreateZerosList()331 int64 leading_dim = input_list_shape.dim_size(0); in GetInitializedList() local333 output_list_shape.InsertDim(0, leading_dim); in GetInitializedList()
413 int64 leading_dim = 0; in Compute() local439 leading_dim += dim; in Compute()442 output_shape.InsertDim(0, leading_dim); in Compute()