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