Searched defs:dim1_size (Results 1 – 4 of 4) sorted by relevance
55 const string& tensor_name, int dim1_size, in CopySubArrayToArray()
1049 int64 dim1_size = values.size(); in CreateR3Projected() local1074 int64 dim1_size = projection_z; in CreateR4Projected() local1183 const int64 dim1_size = values.begin()->size(); in PopulateR2() local
182 int dim1_size = 1; in MaybeMergeConstNodes() local
267 const int64 dim1_size = 1024; in XLA_TEST_F() local