Searched refs:program_shape_proto (Results 1 – 3 of 3) sorted by relevance
186 ProgramShape::ProgramShape(const ProgramShapeProto& program_shape_proto) { in ProgramShape() argument187 for (const ShapeProto& shape_proto : program_shape_proto.parameters()) { in ProgramShape()190 *mutable_result() = Shape(program_shape_proto.result()); in ProgramShape()191 for (const string& name : program_shape_proto.parameter_names()) { in ProgramShape()
225 explicit ProgramShape(const ProgramShapeProto& program_shape_proto);
837 xla::ProgramShapeProto program_shape_proto; in TEST() local838 EXPECT_TRUE(program_shape_proto.ParseFromString(outputs[0].vec<string>()(0))); in TEST()839 xla::ProgramShape program_shape(program_shape_proto); in TEST()