Searched refs:CreateFromShape (Results 1 – 22 of 22) sorted by relevance
72 HloInstruction::CreateConstant(Literal::CreateFromShape(rhs_shape))); in TEST_F()109 HloInstruction::CreateConstant(Literal::CreateFromShape(rhs_shape))); in TEST_F()153 HloInstruction::CreateConstant(Literal::CreateFromShape(rhs_shape))); in TEST_F()189 HloInstruction::CreateConstant(Literal::CreateFromShape(lhs_shape))); in TEST_F()224 HloInstruction::CreateConstant(Literal::CreateFromShape(constant_shape))); in TEST_F()279 HloInstruction::CreateConstant(Literal::CreateFromShape(dot_rhs_shape))); in RunDotOutputFusion()
48 HloInstruction::CreateConstant(Literal::CreateFromShape(shape)))); in Run()
133 HloInstruction::CreateConstant(Literal::CreateFromShape(tuple_shape))); in TEST_F()
466 Literal fake_input = Literal::CreateFromShape(while_cond_param->shape()); in ComputeWhileLoopTripCountUpperBound()
462 HloInstruction::CreateConstant(Literal::CreateFromShape(range))); in CreateComputationWithSignature()
92 HloInstruction::CreateConstant(Literal::CreateFromShape(root_shape)));
1059 auto literal = Literal::CreateFromShape(Shape(arg->shape_with_layout())); in TransferFromOutfeed()
1048 Literal result = Literal::CreateFromShape(gather->shape()); in HandleGather()
3241 auto output = Literal::CreateFromShape(custom_call->shape()); in TEST_F()
2601 Literal::CreateFromShape(reshaped_shape)); in HandleReshape()
506 auto x = Literal::CreateFromShape(tuple.shape()); in TEST_F()976 auto source = Literal::CreateFromShape(shape); in TEST_F()987 auto blank = Literal::CreateFromShape(shape); in TEST_F()1030 const auto const_empty = Literal::CreateFromShape(empty_r1_shape); in TEST_F()1034 const auto empty = Literal::CreateFromShape(empty_r1_shape); in TEST_F()1043 auto empty = Literal::CreateFromShape(empty_r1_shape); in TEST_F()1139 Literal m1 = Literal::CreateFromShape(ShapeUtil::MakeShape(F16, {2, 2})); in TEST_F()1777 Literal scalar_f32 = Literal::CreateFromShape(ShapeUtil::MakeShape(F32, {})); in TEST_F()1781 Literal vector_s32 = Literal::CreateFromShape(ShapeUtil::MakeShape(S32, {3})); in TEST_F()1787 Literal tuple = Literal::CreateFromShape(ShapeUtil::MakeTupleShape( in TEST_F()
87 return Literal::CreateFromShape( in CreateFromDimensions()
318 static Literal CreateFromShape(const Shape& shape);
187 Literal LiteralBase::CreateFromShape(const Shape& shape) { in CreateFromShape() function in xla::LiteralBase595 Literal result = CreateFromShape(shape_with_layout); in Relayout()
149 stream_ptr.get(), Literal::CreateFromShape(lhs_arg_shape), lhs_arg)); in ExecuteAndFetchProfile()156 stream_ptr.get(), Literal::CreateFromShape(rhs_arg_shape), rhs_arg)); in ExecuteAndFetchProfile()
208 return Literal::CreateFromShape(shape); in MakeFakeLiteralInternal()301 return Literal::CreateFromShape(shape); in MakeFakeLiteralInternalWithBounds()
247 auto empty = Literal::CreateFromShape(in_shape); in XLA_TEST_F()
574 auto literal = Literal::CreateFromShape(expected.shape()); in XLA_TEST_F()
68 auto output = Literal::CreateFromShape(custom_call->shape()); in HandleEvaluatorCustomCall()
124 xla::Literal empty_literal = xla::Literal::CreateFromShape( in Compile()
302 ctx->builder(), xla::Literal::CreateFromShape(data_shape)); in CalculateGradData()
303 auto literal = Literal::CreateFromShape(shape); in TransferFromOutfeedLocal()