Lines Matching refs:shapes
64 OpInputList shapes;
65 OP_REQUIRES_OK(context, context->input_list("shapes", &shapes));
66 OP_REQUIRES(context, shapes.size() == N,
67 errors::InvalidArgument("Expected ", N, " input shapes, got ",
68 shapes.size()));
70 OP_REQUIRES(context, TensorShapeUtils::IsVector(shapes[i].shape()),
72 "Input shapes should be a vector but received shape ",
73 shapes[i].shape().DebugString(), " at position ", i));
76 const TensorShape input_shape(shapes[0].vec<int64>());
86 const TensorShape current_shape(shapes[i].vec<int64>());
97 "Input shapes must match: expected ", input_shape.dim_size(j),
126 const TensorShape current_shape(shapes[i].vec<int64>());
