/external/tensorflow/tensorflow/compiler/xla/tests/ |
D | broadcast_simple_test.cc | 293 ConstantLiteral(&b, LiteralUtil::CreateR3<float>( in XLA_TEST_F() 298 LiteralUtil::CreateR3<float>({{{3.0, 7.0}, {4.0, 8.0}, {5.0, 9.0}}, in XLA_TEST_F() 392 LiteralUtil::CreateR3<float>({{{2, 3}, {4, 5}}, {{7, 8}, {9, 10}}}); in XLA_TEST_F() 400 auto r1 = ConstantLiteral(&b, LiteralUtil::CreateR3<float>({{{1, 2}}})); in XLA_TEST_F() 402 &b, LiteralUtil::CreateR3<float>({{{1, 2}, {3, 4}}, {{5, 6}, {7, 8}}})); in XLA_TEST_F() 406 LiteralUtil::CreateR3<float>({{{2, 4}, {4, 6}}, {{6, 8}, {8, 10}}}); in XLA_TEST_F() 413 auto r1 = ConstantLiteral(&b, LiteralUtil::CreateR3<float>({{{1}, {2}}})); in XLA_TEST_F() 415 &b, LiteralUtil::CreateR3<float>({{{1, 2}, {3, 4}}, {{5, 6}, {7, 8}}})); in XLA_TEST_F() 419 LiteralUtil::CreateR3<float>({{{2, 3}, {5, 6}}, {{6, 7}, {9, 10}}}); in XLA_TEST_F() 427 ConstantLiteral(&b, LiteralUtil::CreateR3<float>({{{1, 2}, {3, 4}}})); in XLA_TEST_F() [all …]
|
D | scatter_test.cc | 341 Literal updates = LiteralUtil::CreateR3<int32>( in XLA_TEST_F() 368 LiteralUtil::CreateR3<int32>({{{-1, 1}, {-2, 2}, {-3, 3}}, // in XLA_TEST_F() 398 LiteralUtil::CreateR3<int32>({{{-1, 1}, {-2, 2}, {-3, 3}}, // in XLA_TEST_F() 458 Literal updates = LiteralUtil::CreateR3<int32>({{{10}}, {{20}}}); in XLA_TEST_F() 513 LiteralUtil::CreateR3<int32>({{{0}, {1}}, {{2}, {1}}}); in XLA_TEST_F() 543 Literal updates = LiteralUtil::CreateR3<int32>( in XLA_TEST_F() 573 Literal updates = LiteralUtil::CreateR3<int32>( in XLA_TEST_F() 603 Literal updates = LiteralUtil::CreateR3<int32>( in XLA_TEST_F() 630 LiteralUtil::CreateR3<int32>({{{-1, 1}, {-2, 2}, {-3, 3}}, // in XLA_TEST_F() 634 Literal updates = LiteralUtil::CreateR3<int32>({{{-10, 10}, {-40, 40}}}); in XLA_TEST_F() [all …]
|
D | multioutput_fusion_test.cc | 314 LiteralUtil::CreateR3<float>({{{1, 2}, {3, 4}}, {{5, 6}, {7, 8}}}); in XLA_TEST_F() 343 LiteralUtil::CreateR3<float>({{{1, 2}, {3, 4}}, {{5, 6}, {7, 8}}}); in XLA_TEST_F() 373 LiteralUtil::CreateR3<float>({{{1, 2}, {3, 4}}, {{5, 6}, {7, 8}}}); in XLA_TEST_F() 403 LiteralUtil::CreateR3<float>({{{1, 2}, {3, 4}}, {{5, 6}, {7, 8}}}); in XLA_TEST_F() 407 LiteralUtil::CreateR3<float>({{{1, 2}, {3, 4}}, {{5, 6}, {7, 8}}}), in XLA_TEST_F() 434 LiteralUtil::CreateR3<float>({{{1, 2}, {3, 4}}, {{5, 6}, {7, 8}}}); in XLA_TEST_F() 439 LiteralUtil::CreateR3<float>( in XLA_TEST_F() 467 LiteralUtil::CreateR3<float>({{{1, 2}, {3, 4}}, {{5, 6}, {7, 8}}}); in XLA_TEST_F() 472 LiteralUtil::CreateR3<float>( in XLA_TEST_F() 474 LiteralUtil::CreateR3<float>( in XLA_TEST_F() [all …]
|
D | gather_operation_test.cc | 127 LiteralUtil::CreateR3<int32>({{{0, 2}, {2, 1}}, {{1, 2}, {2, 0}}}); in XLA_TEST_F() 149 LiteralUtil::CreateR3<int32>({{{0, 2}, {2, 1}}, {{1, 2}, {2, 0}}}); in XLA_TEST_F() 169 LiteralUtil::CreateR3<int32>({{{-1, 1}, {-2, 2}, {-3, 3}}, // in XLA_TEST_F() 192 LiteralUtil::CreateR3<int32>({{{-1, 1}, {-2, 2}, {-3, 3}}, // in XLA_TEST_F() 380 Literal operand = LiteralUtil::CreateR3<int32>( in XLA_TEST_F() 496 LiteralUtil::CreateR3<int32>({{{0, 2}, {2, 1}}, {{1, 2}, {2, 0}}}); in XLA_TEST_F() 519 LiteralUtil::CreateR3<int32>({{{-1, 1}, {-2, 2}, {-3, 3}}, // in XLA_TEST_F() 546 LiteralUtil::CreateR3<int32>({{{-1, 1}, {-2, 2}, {-3, 3}}, // in XLA_TEST_F()
|
D | literal_test_util.h | 179 EXPECT_TRUE(Equal(LiteralUtil::CreateR3<NativeT>(expected), actual)); in ExpectR3Equal() 226 EXPECT_TRUE(Near(LiteralUtil::CreateR3<NativeT>(expected), actual, error)); in ExpectR3Near()
|
D | round_trip_transfer_test.cc | 97 LiteralUtil::CreateR3<float>({{{1.0, 2.0}, {1.0, 2.0}, {1.0, 2.0}}, in TEST_F()
|
D | fusion_test.cc | 357 LiteralUtil::CreateR3<int32>({{{1, 2, 3}, {4, 5, 6}}}), in XLA_TEST_F() 365 LiteralUtil::CreateR3<int32>({{{1, 2, 3}, {4, 5, 6}}}))); in XLA_TEST_F() 380 HloInstruction::CreateConstant(LiteralUtil::CreateR3<int32>({{{7}}}))); in XLA_TEST_F() 402 LiteralTestUtil::Equal(LiteralUtil::CreateR3<int32>({{{7}}}), in XLA_TEST_F()
|
D | copy_test.cc | 73 TestCopyOp(LiteralUtil::CreateR3({{{1.0f, 2.0f, 3.0f}, {4.0f, 5.0f, 6.0f}}, in XLA_TEST_F()
|
D | broadcast_test.cc | 146 LiteralUtil::CreateR3<float>({{{1.0, 2.0}, {1.0, 2.0}, {1.0, 2.0}}, in XLA_TEST_F()
|
D | reshape_test.cc | 468 auto expected_literal = LiteralUtil::CreateR3<float>( in XLA_TEST_P()
|
/external/tensorflow/tensorflow/compiler/xla/service/ |
D | elemental_ir_emitter_test.cc | 59 Literal lhs = LiteralUtil::CreateR3<int32>({{{1}, {2}}}); in XLA_TEST_F() 60 Literal rhs = LiteralUtil::CreateR3<int32>({{{3}, {4}}}); in XLA_TEST_F()
|
D | hlo_creation_utils_test.cc | 80 evaluator.Evaluate(*module, {LiteralUtil::CreateR3<int32>( in TEST_F() 123 CHECK_EQ(result_literal, LiteralUtil::CreateR3<int32>({{{9, 10}}})); in TEST_F() 163 LiteralUtil::CreateR3<int32>({{{1, 2}}, {{3, 4}}, {{5, 6}}})); in TEST_F()
|
D | hlo_evaluator_test.cc | 444 auto output_literal = LiteralUtil::CreateR3<int32>( in TEST_F() 2157 LiteralUtil::CreateR3<int32>( in TEST_F() 2179 LiteralUtil::CreateR3<int32>({{{-1, 1}, {-2, 2}, {-3, 3}}, // in TEST_F() 2206 LiteralUtil::CreateR3<int32>({{{-1, 1}, {-2, 2}, {-3, 3}}, // in TEST_F() 2260 LiteralUtil::CreateR3<int32>({{{8}}, {{5}}}), result)); in TEST_F() 2305 LiteralUtil::CreateR3<int32>({{{0}, {1}}, {{2}, {1}}}); in TEST_F() 2542 Literal updates = LiteralUtil::CreateR3<int32>( in TEST_F() 2574 LiteralUtil::CreateR3<int32>({{{-1, 1}, {-2, 2}, {-3, 3}}, // in TEST_F() 2580 LiteralUtil::CreateR3<int32>({{{-10, 10}, {-2, 2}, {-3, 3}}, // in TEST_F() 2612 LiteralUtil::CreateR3<int32>({{{-1, 1}, {-2, 2}, {-3, 3}}, // in TEST_F() [all …]
|
/external/tensorflow/tensorflow/compiler/xla/service/gpu/tests/ |
D | infeed_test.cc | 68 LiteralUtil::CreateR3({{{1.0f, 2.0f, 3.0f}, {4.0f, 5.0f, 6.0f}}, in TEST_F()
|
/external/tensorflow/tensorflow/compiler/xla/ |
D | literal_util.h | 84 static Literal CreateR3(std::initializer_list< 375 /* static */ Literal LiteralUtil::CreateR3( in CreateR3() function
|
D | literal_test.cc | 154 LiteralUtil::CreateR3({{{1}, {2}}, {{3}, {4}}, {{5}, {6}}}); in TEST_F() 830 auto input_2x3x2 = LiteralUtil::CreateR3<uint32>( in TEST_F() 961 auto expected = LiteralUtil::CreateR3<uint32>( in TEST_F()
|
/external/tensorflow/tensorflow/compiler/xla/service/cpu/tests/ |
D | cpu_infeed_test.cc | 74 LiteralUtil::CreateR3({{{1.0f, 2.0f, 3.0f}, {4.0f, 5.0f, 6.0f}}, in TEST_F()
|