Searched refs:set_num_elements (Results 1 – 2 of 2) sorted by relevance
125 set_num_elements(-1); in TensorShapeBase()128 set_num_elements(1); in TensorShapeBase()183 set_num_elements(neg ? -1 : size); in InitDims()192 set_num_elements(neg ? -1 : (size0 * size1)); in InitDims()203 set_num_elements(neg ? -1 : (size0 * size1 * size2)); in InitDims()216 set_num_elements(neg ? -1 : (size0 * size1 * size2 * size3)); in InitDims()223 set_num_elements(1); in InitDims()235 set_num_elements(-1); in TensorShapeBase()238 set_num_elements(1); in TensorShapeBase()301 set_num_elements(1); in ClearAllButDataType()[all …]
144 void set_num_elements(int64 n) { num_elements_ = n; } in set_num_elements() function