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

  /external/tensorflow/tensorflow/compiler/xla/tests/
deep_graph_test.cc 29 auto x_data = CreateR0Parameter<int32>(3, 0, "x", &b, &x);
30 auto y_data = CreateR0Parameter<int32>(1, 1, "y", &b, &y);
conditional_test.cc 182 auto pred_arg = CreateR0Parameter<bool>(true, 0, "pred", &builder, &pred);
198 auto branch_index_arg = CreateR0Parameter<int32>(bi, 0, "branch_index_arg",
225 auto pred_arg = CreateR0Parameter<bool>(false, 0, "pred", &builder, &pred);
241 auto branch_index_arg = CreateR0Parameter<int32>(bi, 0, "branch_index_arg",
278 auto pred_arg = CreateR0Parameter<bool>(false, 0, "pred", &builder, &pred);
292 auto pred_arg = CreateR0Parameter<bool>(false, 0, "pred", &builder, &pred);
305 auto pred_arg = CreateR0Parameter<bool>(false, 0, "pred", &builder, &pred);
319 auto pred_arg = CreateR0Parameter<bool>(false, 0, "pred", &builder, &pred);
332 auto pred_arg = CreateR0Parameter<bool>(false, 0, "pred", &builder, &pred);
354 auto pred_arg = CreateR0Parameter<bool>(false, 0, "pred", &builder, &pred)
    [all...]
dynamic_ops_test.cc 138 std::unique_ptr<GlobalData> start_data = CreateR0Parameter<IndexT>(
166 start_data[i] = CreateR0Parameter<IndexT>(
201 start_data[i] = CreateR0Parameter<IndexT>(
421 std::unique_ptr<GlobalData> start_data = CreateR0Parameter<IndexT>(
454 start_data[i] = CreateR0Parameter<IndexT>(
493 start_data[i] = CreateR0Parameter<IndexT>(
    [all...]
scalar_computations_test.cc 302 CreateR0Parameter<int32>(p.dividend, 0, "dividend", &builder, &dividend);
304 CreateR0Parameter<int32>(p.divisor, 1, "divisor", &builder, &divisor);
317 CreateR0Parameter<int32>(p.dividend, 0, "dividend", &builder, &dividend);
319 CreateR0Parameter<int32>(p.divisor, 1, "divisor", &builder, &divisor);
tuple_test.cc 211 CreateR0Parameter<float>(0.0f, /*parameter_number=*/0, /*name=*/"v1",
214 CreateR0Parameter<float>(1.0f, /*parameter_number=*/1, /*name=*/"v2",
client_library_test_base.h 327 std::unique_ptr<GlobalData> CreateR0Parameter(NativeT value,
549 std::unique_ptr<GlobalData> ClientLibraryTestBase::CreateR0Parameter(
broadcast_simple_test.cc 125 CreateR0Parameter<float>(2.25f, /*parameter_number=*/0, /*name=*/"src",
  /external/tensorflow/tensorflow/compiler/xla/client/lib/
slicing_test.cc 62 auto x_data = CreateR0Parameter<int>(2, 1, "x", &builder, &x);
63 auto y_data = CreateR0Parameter<int>(1, 2, "y", &builder, &y);
77 auto index_data = CreateR0Parameter<int>(1, 1, "index", &builder, &index);
92 auto x_data = CreateR0Parameter<int>(2, 2, "x", &builder, &x);
93 auto y_data = CreateR0Parameter<int>(1, 3, "y", &builder, &y);
matrix_test.cc 108 auto index_data = CreateR0Parameter<int>(1, 2, "index", &builder, &index);
129 auto index_data = CreateR0Parameter<int>(1, 2, "index", &builder, &index);

Completed in 112 milliseconds