Searched refs:leading_dim_dynamic_size (Results 1 – 2 of 2) sorted by relevance
207 xla::XlaOp* leading_dim_dynamic_size) { in GetLeadingDimForTensorList() argument216 *leading_dim_dynamic_size = xla::GetDimensionSize(buffer, 0); in GetLeadingDimForTensorList()220 *leading_dim_dynamic_size = xla::GetDimensionSize(list, 0); in GetLeadingDimForTensorList()300 xla::XlaOp leading_dim_dynamic_size; in GetInitializedTensorListForElement() local303 list, &leading_dim, &leading_dim_is_dynamic, &leading_dim_dynamic_size)); in GetInitializedTensorListForElement()335 dynamic_dims.push_back(leading_dim_dynamic_size); in GetInitializedTensorListForElement()
72 xla::XlaOp* leading_dim_dynamic_size);