Searched defs:TensorShapeDim (Results 1 – 1 of 1) sorted by relevance
451 struct TensorShapeDim { struct452 explicit TensorShapeDim(int64_t s) : size(s) {} in TensorShapeDim() function470 TensorShapeDim operator*() { return TensorShapeDim(shape_->dim_size(d_)); } argument