Searched refs:dim_handle (Results 1 – 2 of 2) sorted by relevance
696 auto dim_handle = c->Dim(c->input(i), 0); in __anondee5e0a21402() local697 if (c->ValueKnown(dim_handle)) { in __anondee5e0a21402()699 dim_zero = c->Value(dim_handle); in __anondee5e0a21402()700 } else if (c->Value(dim_handle) != dim_zero) { in __anondee5e0a21402()
619 const shape_inference::DimensionHandle dim_handle = in IsNodeFlattenReshape() local623 if (context->ValueKnown(dim_handle)) { in IsNodeFlattenReshape()