Searched refs:in_shapes (Results 1 – 2 of 2) sorted by relevance
58 std::vector<shape_inference::ShapeHandle> in_shapes; in __anon720d5c150102() local59 TF_RETURN_IF_ERROR(c->input("in_tensors", &in_shapes)); in __anon720d5c150102()60 std::vector<shape_inference::ShapeHandle> out_shapes(in_shapes.size()); in __anon720d5c150102()61 for (int i = 0; i < in_shapes.size(); ++i) { in __anon720d5c150102()63 c->ReplaceDim(in_shapes[i], 0, c->UnknownDim(), &out_shapes[i])); in __anon720d5c150102()
39 std::vector<ShapeHandle> in_shapes; in InferShapes() local43 in_shapes.push_back(shape); in InferShapes()64 op.graph_def_version, &op.node_def, op_reg_data->op_def, in_shapes, in InferShapes()