Searched defs:leading_dim_size (Results 1 – 3 of 3) sorted by relevance
524 xla::XlaOp leading_dim_size = xla::GetDimensionSize(input, 0); in Compile() local
194 xla::XlaOp leading_dim_size) { in BuildUninitializedTensorList()
93 xla::XlaOp leading_dim_size; in Compile() local