HomeSort by relevance Sort by last modified time
    Searched defs:arg_literal (Results 1 - 3 of 3) sorted by null

  /external/tensorflow/tensorflow/compiler/xla/tests/
reduce_window_test.cc 317 std::unique_ptr<Literal> arg_literal = Literal::CreateFromShape(shape); local
321 TF_EXPECT_OK(arg_literal->Populate<float>(generator));
323 const auto input = CreateConstantFromLiteral(*arg_literal, &builder_);
346 std::unique_ptr<Literal> arg_literal = local
349 const auto input = CreateConstantFromLiteral(*arg_literal, &builder_);
    [all...]
local_client_execute_test.cc 360 auto arg_literal = Literal::MakeTuple( local
366 auto arg_buffer = LiteralToShapedBuffer(*arg_literal);
394 auto arg_literal = Literal::MakeTuple( local
397 auto arg_buffer = LiteralToShapedBuffer(*arg_literal);
452 std::unique_ptr<Literal> arg_literal = local
454 auto arg_buffer = LiteralToShapedBuffer(*arg_literal);
511 auto arg_literal = Literal::MakeTupleOwned(std::move(outer_tuple_elements)); local
512 auto arg_buffer = LiteralToShapedBuffer(*arg_literal);
551 std::unique_ptr<Literal> arg_literal = Literal::CreateR0<float>(123.0); local
554 arg_vector.push_back(std::move(arg_literal));
    [all...]
  /external/tensorflow/tensorflow/compiler/xla/service/
hlo_evaluator_test.cc 1218 auto arg_literal = Literal::CreateR2FromArray2D<float>(*arg_array); local
1260 auto arg_literal = Literal::CreateR2FromArray2D<float>(*arg_array); local
1311 auto arg_literal = Literal::CreateR2FromArray2D<float>(*arg_array); local
1363 std::unique_ptr<Literal> arg_literal = local
    [all...]

Completed in 289 milliseconds