Searched refs:program_shape_proto (Results 1 – 4 of 4) sorted by relevance
222 ProgramShape::ProgramShape(const ProgramShapeProto& program_shape_proto) { in ProgramShape() argument223 for (const ShapeProto& shape_proto : program_shape_proto.parameters()) { in ProgramShape()226 *mutable_result() = Shape(program_shape_proto.result()); in ProgramShape()227 for (const std::string& name : program_shape_proto.parameter_names()) { in ProgramShape()
318 explicit ProgramShape(const ProgramShapeProto& program_shape_proto);
859 xla::ProgramShapeProto program_shape_proto; in Compile() local861 compiled_program->program_shape(&program_shape_proto); in Compile()868 result_layout = ::xla::Shape(program_shape_proto.result()); in Compile()
1548 xla::ProgramShapeProto program_shape_proto; in TEST() local1550 ParseFromTString(outputs[0].vec<tstring>()(0), &program_shape_proto)); in TEST()1551 xla::ProgramShape program_shape(program_shape_proto); in TEST()