Searched refs:CreateR2WithLayout (Results 1 – 14 of 14) sorted by relevance
141 auto rowmaj_array = LiteralUtil::CreateR2WithLayout( in XLA_TEST_F()145 auto colmaj_array = LiteralUtil::CreateR2WithLayout( in XLA_TEST_F()
465 Literal literal = LiteralUtil::CreateR2WithLayout<float>( in XLA_TEST_F()476 Literal literal = LiteralUtil::CreateR2WithLayout<float>( in XLA_TEST_F()
369 Literal param0_literal = LiteralUtil::CreateR2WithLayout( in XLA_TEST_F()374 Literal param1_literal = LiteralUtil::CreateR2WithLayout( in XLA_TEST_F()
58 Literal expected_literal = LiteralUtil::CreateR2WithLayout<int32_t>( in XLA_TEST_F()
259 Literal expected_literal = LiteralUtil::CreateR2WithLayout<int32_t>( in XLA_TEST_F()
132 auto x_array = LiteralToShapedBuffer(LiteralUtil::CreateR2WithLayout( in XLA_TEST_F()138 auto y_array = LiteralToShapedBuffer(LiteralUtil::CreateR2WithLayout( in XLA_TEST_F()
197 Literal literal = LiteralUtil::CreateR2WithLayout<float>( in XLA_TEST_F()
548 ->TransferToServer(LiteralUtil::CreateR2WithLayout<complex64>( in XLA_TEST_F()553 ->TransferToServer(LiteralUtil::CreateR2WithLayout<complex64>( in XLA_TEST_F()
85 static Literal CreateR2WithLayout(305 /* static */ Literal LiteralUtil::CreateR2WithLayout( in CreateR2WithLayout() function320 return CreateR2WithLayout(values, LayoutUtil::GetDefaultLayoutForR2()); in CreateR2()
789 auto mat_dim0minor = LiteralUtil::CreateR2WithLayout<int32_t>( in TEST_F()800 auto mat_dim0major = LiteralUtil::CreateR2WithLayout<int32_t>( in TEST_F()
166 auto constant_literal1 = LiteralUtil::CreateR2WithLayout<float>( in TEST_F()168 auto constant_literal2 = LiteralUtil::CreateR2WithLayout<float>( in TEST_F()217 HloInstruction::CreateConstant(LiteralUtil::CreateR2WithLayout<float>( in TEST_F()220 HloInstruction::CreateConstant(LiteralUtil::CreateR2WithLayout<float>( in TEST_F()813 HloInstruction::CreateConstant(LiteralUtil::CreateR2WithLayout<float>( in TEST_F()
85 HloInstruction::CreateConstant(LiteralUtil::CreateR2WithLayout<float>( in TEST_F()88 HloInstruction::CreateConstant(LiteralUtil::CreateR2WithLayout<float>( in TEST_F()
656 auto input_literal = LiteralUtil::CreateR2WithLayout<int32_t>( in TEST_P()658 auto expected = LiteralUtil::CreateR2WithLayout<float>( in TEST_P()
209 auto array = xla::LiteralUtil::CreateR2WithLayout<float>(v, layout); in FloatMatrix()1621 xla::LiteralUtil::CreateR2WithLayout<float>({{18.0f}, {44.0f}}, layout); in TEST()