Searched refs:set_ndims_byte (Results 1 – 2 of 2) sorted by relevance
147 set_ndims_byte(kUnknownRank); in TensorShapeBase()150 set_ndims_byte(0); in TensorShapeBase()167 out->set_ndims_byte(kUnknownRank); in BuildTensorShapeBase()170 out->set_ndims_byte(0); in BuildTensorShapeBase()243 set_ndims_byte(1); in InitDims()250 set_ndims_byte(2); in InitDims()259 set_ndims_byte(3); in InitDims()270 set_ndims_byte(4); in InitDims()285 set_ndims_byte(0); in InitDims()303 set_ndims_byte(kUnknownRank); in TensorShapeBase()[all …]
139 void set_ndims_byte(uint8 nd) { buf()[14] = nd; } in set_ndims_byte() function706 set_ndims_byte(1); in TensorShapeBase()