Searched refs:min_shape (Results 1 – 3 of 3) sorted by relevance
30 const TensorShape min_shape = ctx->InputShape(1); in Compile() local42 " clip_value_min shape: ", min_shape.DebugString(), in Compile()46 if (shape != min_shape) { in Compile()47 OP_REQUIRES(ctx, TensorShapeUtils::IsScalar(min_shape), shape_error()); in Compile()
336 ShapeHandle min_shape; in __anon30b5031d0302() local337 TF_RETURN_IF_ERROR(c->WithRank(c->input(2), 0, &min_shape)); in __anon30b5031d0302()
1247 TF_ASSIGN_OR_RETURN(Shape min_shape, compact_shape_function_(original_shape)); in GetCompactShape()1248 compact_shape_[hlo] = min_shape; in GetCompactShape()1249 return min_shape; in GetCompactShape()