Searched refs:program_shape_proto (Results 1 – 4 of 4) sorted by relevance
189 ProgramShape::ProgramShape(const ProgramShapeProto& program_shape_proto) { in ProgramShape() argument190 for (const ShapeProto& shape_proto : program_shape_proto.parameters()) { in ProgramShape()193 *mutable_result() = Shape(program_shape_proto.result()); in ProgramShape()194 for (const string& name : program_shape_proto.parameter_names()) { in ProgramShape()
260 explicit ProgramShape(const ProgramShapeProto& program_shape_proto);
821 xla::ProgramShapeProto program_shape_proto; in Compile() local823 compiled_program->program_shape(&program_shape_proto); in Compile()830 result_layout = ::xla::Shape(program_shape_proto.result()); in Compile()
1133 xla::ProgramShapeProto program_shape_proto; in TEST() local1135 program_shape_proto.ParseFromString(outputs[0].vec<tstring>()(0))); in TEST()1136 xla::ProgramShape program_shape(program_shape_proto); in TEST()