Home
last modified time | relevance | path

Searched refs:CreateR3 (Results 1 – 17 of 17) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/tests/
Dbroadcast_simple_test.cc293 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 …]
Dscatter_test.cc341 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 …]
Dmultioutput_fusion_test.cc314 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 …]
Dgather_operation_test.cc127 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()
Dliteral_test_util.h179 EXPECT_TRUE(Equal(LiteralUtil::CreateR3<NativeT>(expected), actual)); in ExpectR3Equal()
226 EXPECT_TRUE(Near(LiteralUtil::CreateR3<NativeT>(expected), actual, error)); in ExpectR3Near()
Dround_trip_transfer_test.cc97 LiteralUtil::CreateR3<float>({{{1.0, 2.0}, {1.0, 2.0}, {1.0, 2.0}}, in TEST_F()
Dfusion_test.cc357 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()
Dcopy_test.cc73 TestCopyOp(LiteralUtil::CreateR3({{{1.0f, 2.0f, 3.0f}, {4.0f, 5.0f, 6.0f}}, in XLA_TEST_F()
Dbroadcast_test.cc146 LiteralUtil::CreateR3<float>({{{1.0, 2.0}, {1.0, 2.0}, {1.0, 2.0}}, in XLA_TEST_F()
Dreshape_test.cc468 auto expected_literal = LiteralUtil::CreateR3<float>( in XLA_TEST_P()
/external/tensorflow/tensorflow/compiler/xla/service/
Delemental_ir_emitter_test.cc59 Literal lhs = LiteralUtil::CreateR3<int32>({{{1}, {2}}}); in XLA_TEST_F()
60 Literal rhs = LiteralUtil::CreateR3<int32>({{{3}, {4}}}); in XLA_TEST_F()
Dhlo_creation_utils_test.cc80 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()
Dhlo_evaluator_test.cc444 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/
Dinfeed_test.cc68 LiteralUtil::CreateR3({{{1.0f, 2.0f, 3.0f}, {4.0f, 5.0f, 6.0f}}, in TEST_F()
/external/tensorflow/tensorflow/compiler/xla/
Dliteral_util.h84 static Literal CreateR3(std::initializer_list<
375 /* static */ Literal LiteralUtil::CreateR3( in CreateR3() function
Dliteral_test.cc154 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/
Dcpu_infeed_test.cc74 LiteralUtil::CreateR3({{{1.0f, 2.0f, 3.0f}, {4.0f, 5.0f, 6.0f}}, in TEST_F()