HomeSort by relevance Sort by last modified time
    Searched refs:CreateR2FromArray2D (Results 1 - 19 of 19) sorted by null

  /external/tensorflow/tensorflow/compiler/xla/service/cpu/tests/
cpu_external_constants_test.cc 43 LiteralUtil::CreateR2FromArray2D(backing_array)));
  /external/tensorflow/tensorflow/compiler/xla/
reference_util_test.cc 56 auto actual_literal = LiteralUtil::CreateR2FromArray2D(*result);
68 auto actual_literal = LiteralUtil::CreateR2FromArray2D(*result);
99 auto actual_literal = LiteralUtil::CreateR2FromArray2D(*result);
109 auto actual_literal = LiteralUtil::CreateR2FromArray2D(*result);
147 auto actual_literal = LiteralUtil::CreateR2FromArray2D(*result);
155 auto actual_literal = LiteralUtil::CreateR2FromArray2D(*result);
425 auto actual_literal = LiteralUtil::CreateR2FromArray2D(*actual);
literal_util.h 169 static Literal CreateR2FromArray2D(const Array2D<NativeT>& values);
459 /* static */ Literal LiteralUtil::CreateR2FromArray2D(
548 return CreateR2FromArray2D(array);
literal_util.cc 289 return CreateR2FromArray2D(*value);
  /external/tensorflow/tensorflow/compiler/xla/tests/
matrix_ops_simple_test.cc 67 LiteralUtil::CreateR2FromArray2D<T>({{2.71828f, 1.00000f}, // row 0
96 LiteralUtil::CreateR2FromArray2D<T>({{1.5f, 0.5f}, // row 0
115 LiteralUtil::CreateR2FromArray2D<T>({{7.0f, 6.0f}, // row 0
round_trip_transfer_test.cc 88 RoundTripTest(LiteralUtil::CreateR2FromArray2D<float>(Array2D<float>(10, 0)));
reduce_test.cc 179 Literal input_literal = LiteralUtil::CreateR2FromArray2D(input_data);
212 Literal input_literal = LiteralUtil::CreateR2FromArray2D(input_data);
239 Literal input_literal = LiteralUtil::CreateR2FromArray2D(input_data);
296 Literal input_literal = LiteralUtil::CreateR2FromArray2D(input_data);
450 Literal input_literal = LiteralUtil::CreateR2FromArray2D(input_data);
481 Literal input_literal = LiteralUtil::CreateR2FromArray2D(input_data);
590 auto input_literal = LiteralUtil::CreateR2FromArray2D(input);
605 auto input_literal = LiteralUtil::CreateR2FromArray2D(input);
619 auto input_literal = LiteralUtil::CreateR2FromArray2D(input);
631 auto input_literal = LiteralUtil::CreateR2FromArray2D(input)
    [all...]
custom_call_test.cc 107 HloInstruction::CreateConstant(LiteralUtil::CreateR2FromArray2D(array)));
122 HloInstruction::CreateConstant(LiteralUtil::CreateR2FromArray2D(
literal_test_util.h 185 EXPECT_TRUE(Equal(LiteralUtil::CreateR2FromArray2D(expected), actual));
242 EXPECT_TRUE(Near(LiteralUtil::CreateR2FromArray2D(expected), actual, error));
reshape_test.cc 58 auto input_literal = LiteralUtil::CreateR2FromArray2D(input_array);
100 auto input_literal = LiteralUtil::CreateR2FromArray2D(input_array);
131 auto input_literal = LiteralUtil::CreateR2FromArray2D(input_array);
145 LiteralUtil::CreateR2FromArray2D<float>(Array2D<float>(0, 3));
158 auto input_literal = LiteralUtil::CreateR2FromArray2D(input_array);
627 Literal expected = LiteralUtil::CreateR2FromArray2D<float>(expected_array);
746 auto expected = LiteralUtil::CreateR2FromArray2D(expected_array);
    [all...]
dot_operation_test.cc 200 ->TransferToServer(LiteralUtil::CreateR2FromArray2D<T>(
204 ->TransferToServer(LiteralUtil::CreateR2FromArray2D<T>(
666 ->TransferToServer(LiteralUtil::CreateR2FromArray2D<T>(
694 ->TransferToServer(LiteralUtil::CreateR2FromArray2D<T>(
842 LiteralUtil::CreateR2FromArray2D<T>(*arg_0_value_array)));
846 LiteralUtil::CreateR2FromArray2D<T>(*arg_1_value_array)));
850 LiteralUtil::CreateR2FromArray2D<T>(*arg_2_value_array)));
    [all...]
dynamic_ops_test.cc 153 std::move(LiteralUtil::CreateR2FromArray2D(input_values_int)
157 std::move(LiteralUtil::CreateR2FromArray2D(expected_values_int)
437 std::move(LiteralUtil::CreateR2FromArray2D(input_values_int)
441 std::move(LiteralUtil::CreateR2FromArray2D(update_values_int)
445 std::move(LiteralUtil::CreateR2FromArray2D(expected_values_int)
    [all...]
broadcast_test.cc 249 LiteralUtil::CreateR2FromArray2D<float>(to_broadcast)));
fusion_test.cc 99 LiteralUtil::CreateR2FromArray2D(operand_data[i])));
130 auto expected = LiteralUtil::CreateR2FromArray2D(answer_data);
    [all...]
client_library_test_base.h 472 LiteralUtil::CreateR2FromArray2D<NativeT>(expected);
489 LiteralUtil::CreateR2FromArray2D<NativeT>(expected);
580 Literal literal = LiteralUtil::CreateR2FromArray2D(array_2d);
broadcast_simple_test.cc 77 auto r2_data = LiteralUtil::CreateR2FromArray2D(*r2_array).Relayout(
615 auto expected = LiteralUtil::CreateR2FromArray2D(expected_array);
params_test.cc 96 LiteralUtil::CreateR2FromArray2D<float>(Array2D<float>(3, 0));
  /external/tensorflow/tensorflow/compiler/xla/service/
hlo_evaluator_test.cc 643 auto input = LiteralUtil::CreateR2FromArray2D<float>(*input_array);
668 auto expected = LiteralUtil::CreateR2FromArray2D<float>(*expected_array);
684 auto input = LiteralUtil::CreateR2FromArray2D<float>(*input_array);
707 auto expected = LiteralUtil::CreateR2FromArray2D<float>(*expected_array);
724 auto lhs_literal = LiteralUtil::CreateR2FromArray2D<float>(*lhs_array);
753 auto expected = LiteralUtil::CreateR2FromArray2D<float>(expected_array);
776 auto rhs_literal = LiteralUtil::CreateR2FromArray2D<float>(*rhs_array);
808 auto lhs_literal = LiteralUtil::CreateR2FromArray2D<float>(*lhs_array);
820 auto rhs_literal = LiteralUtil::CreateR2FromArray2D<float>(*rhs_array);
    [all...]
  /external/tensorflow/tensorflow/compiler/xla/client/
xla_builder.h     [all...]

Completed in 1062 milliseconds