Searched refs:TensorShapeDim (Results 1 – 2 of 2) sorted by relevance
451 struct TensorShapeDim { struct452 explicit TensorShapeDim(int64_t s) : size(s) {} in TensorShapeDim() argument470 TensorShapeDim operator*() { return TensorShapeDim(shape_->dim_size(d_)); } argument
350 for (tensorflow::TensorShapeDim dim : tensor_resource_shape.shape) { in RefineShapes()