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.cc370 Literal updates = LiteralUtil::CreateR3<int32>( in XLA_TEST_F()
397 LiteralUtil::CreateR3<int32>({{{-1, 1}, {-2, 2}, {-3, 3}}, // in XLA_TEST_F()
427 LiteralUtil::CreateR3<int32>({{{-1, 1}, {-2, 2}, {-3, 3}}, // in XLA_TEST_F()
487 Literal updates = LiteralUtil::CreateR3<int32>({{{10}}, {{20}}}); in XLA_TEST_F()
542 LiteralUtil::CreateR3<int32>({{{0}, {1}}, {{2}, {1}}}); in XLA_TEST_F()
572 Literal updates = LiteralUtil::CreateR3<int32>( in XLA_TEST_F()
602 Literal updates = LiteralUtil::CreateR3<int32>( in XLA_TEST_F()
637 Literal updates = LiteralUtil::CreateR3<int32>( in XLA_TEST_F()
664 LiteralUtil::CreateR3<int32>({{{-1, 1}, {-2, 2}, {-3, 3}}, // in XLA_TEST_F()
668 Literal updates = LiteralUtil::CreateR3<int32>({{{-10, 10}, {-40, 40}}}); in XLA_TEST_F()
[all …]
Dgather_operation_test.cc84 LiteralUtil::CreateR3<int32>({{{1, 2, 3}, {4, 5, 6}}, in XLA_TEST_F()
151 LiteralUtil::CreateR3<int32>({{{0, 2}, {2, 1}}, {{1, 2}, {2, 0}}}); in XLA_TEST_F()
173 LiteralUtil::CreateR3<int32>({{{0, 2}, {2, 1}}, {{1, 2}, {2, 0}}}); in XLA_TEST_F()
193 LiteralUtil::CreateR3<int32>({{{-1, 1}, {-2, 2}, {-3, 3}}, // in XLA_TEST_F()
216 LiteralUtil::CreateR3<int32>({{{-1, 1}, {-2, 2}, {-3, 3}}, // in XLA_TEST_F()
473 Literal operand = LiteralUtil::CreateR3<int32>( in XLA_TEST_F()
589 LiteralUtil::CreateR3<int32>({{{0, 2}, {2, 1}}, {{1, 2}, {2, 0}}}); in XLA_TEST_F()
612 LiteralUtil::CreateR3<int32>({{{-1, 1}, {-2, 2}, {-3, 3}}, // in XLA_TEST_F()
639 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()
Dcpu_gpu_fusion_test.cc356 LiteralUtil::CreateR3<int32>({{{1, 2, 3}, {4, 5, 6}}}), in XLA_TEST_F()
364 LiteralUtil::CreateR3<int32>({{{1, 2, 3}, {4, 5, 6}}}))); in XLA_TEST_F()
379 HloInstruction::CreateConstant(LiteralUtil::CreateR3<int32>({{{7}}}))); in XLA_TEST_F()
401 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.cc495 auto expected_literal = LiteralUtil::CreateR3<float>( in XLA_TEST_P()
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_evaluator_test.cc521 auto output_literal = LiteralUtil::CreateR3<int32>( in TEST_F()
1507 fft_c64x2x4x8_ = LiteralUtil::CreateR3<complex64>({ in InitializeFftData()
1525 fft_c64x2x4x8_1d_ = LiteralUtil::CreateR3<complex64>({ in InitializeFftData()
1543 fft_c64x2x4x8_2d_ = LiteralUtil::CreateR3<complex64>({ in InitializeFftData()
1561 fft_c64x2x4x8_3d_ = LiteralUtil::CreateR3<complex64>({ in InitializeFftData()
1987 auto input = LiteralUtil::CreateR3<float>(
1993 auto expected = LiteralUtil::CreateR3<complex64>(
2018 auto input = LiteralUtil::CreateR3<complex64>(
2028 auto expected = LiteralUtil::CreateR3<float>(
2049 auto input = LiteralUtil::CreateR3<float>({{{1.8, 2.7, 3.6, 4.5, 5.4},
[all …]
Delemental_ir_emitter_test.cc70 Literal lhs = LiteralUtil::CreateR3<int32>({{{1}, {2}}}); in XLA_TEST_F()
71 Literal rhs = LiteralUtil::CreateR3<int32>({{{3}, {4}}}); in XLA_TEST_F()
Dhlo_creation_utils_test.cc95 evaluator.Evaluate(*module, {LiteralUtil::CreateR3<int32>( in TEST_F()
138 CHECK_EQ(result_literal, LiteralUtil::CreateR3<int32>({{{9, 10}}})); in TEST_F()
178 LiteralUtil::CreateR3<int32>({{{1, 2}}, {{3, 4}}, {{5, 6}}})); in TEST_F()
Ddynamic_padder_test.cc557 LiteralUtil::CreateR3<int32>({{{1}, {2}}, {{3}, {4}}, {{5}, {6}}}); in XLA_TEST_F()
940 Literal operand = LiteralUtil::CreateR3<float>({{{1, 2, 3, 4, 5}}}); in XLA_TEST_F()
945 Literal expected = LiteralUtil::CreateR3<float>({{{15}}}); in XLA_TEST_F()
1059 Literal operand = LiteralUtil::CreateR3<int32>({{{0, -1}, {1, -1}}, in XLA_TEST_F()
1189 Literal operand = LiteralUtil::CreateR3<int32>( in XLA_TEST_F()
1231 Literal operand = LiteralUtil::CreateR3<int32>( in XLA_TEST_F()
1287 LiteralUtil::CreateR3<int32>({{{0, 1}, {3, 4}}, {{0, 1}, {3, 4}}}); in XLA_TEST_F()
/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/service/cpu/tests/
Dcpu_infeed_test.cc74 LiteralUtil::CreateR3({{{1.0f, 2.0f, 3.0f}, {4.0f, 5.0f, 6.0f}}, in TEST_F()
/external/tensorflow/tensorflow/compiler/xla/
Dliteral_util.h83 static Literal CreateR3(std::initializer_list<
357 /* static */ Literal LiteralUtil::CreateR3( in CreateR3() function
Dliteral_test.cc164 LiteralUtil::CreateR3({{{1}, {2}}, {{3}, {4}}, {{5}, {6}}}); in TEST_F()
867 auto input_2x3x2 = LiteralUtil::CreateR3<uint32>( in TEST_F()
1030 auto expected = LiteralUtil::CreateR3<uint32>( in TEST_F()