Searched refs:program_shape_proto (Results 1 – 4 of 4) sorted by relevance
196 ProgramShape::ProgramShape(const ProgramShapeProto& program_shape_proto) { in ProgramShape() argument197 for (const ShapeProto& shape_proto : program_shape_proto.parameters()) { in ProgramShape()200 *mutable_result() = Shape(program_shape_proto.result()); in ProgramShape()201 for (const string& name : program_shape_proto.parameter_names()) { in ProgramShape()
276 explicit ProgramShape(const ProgramShapeProto& program_shape_proto);
849 xla::ProgramShapeProto program_shape_proto; in Compile() local851 compiled_program->program_shape(&program_shape_proto); in Compile()858 result_layout = ::xla::Shape(program_shape_proto.result()); in Compile()
1541 xla::ProgramShapeProto program_shape_proto; in TEST() local1543 ParseFromTString(outputs[0].vec<tstring>()(0), &program_shape_proto)); in TEST()1544 xla::ProgramShape program_shape(program_shape_proto); in TEST()