Searched refs:pshape (Results 1 – 1 of 1) sorted by relevance
63 xla::ProgramShapeProto* pshape = &compile_result->program_shape; in CompileXla() local68 std::vector<const xla::Shape*> arg_layout_ptrs(pshape->parameters_size()); in CompileXla()69 std::vector<xla::Shape> arg_layouts(pshape->parameters_size()); in CompileXla()70 for (int i = 0; i < pshape->parameters_size(); ++i) { in CompileXla()71 arg_layouts[i] = xla::Shape(*pshape->mutable_parameters(i)); in CompileXla()77 xla::Shape result_shape(pshape->result()); in CompileXla()