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

  /external/tensorflow/tensorflow/compiler/xla/service/
hlo_runner.cc 170 auto result_literal = backend().transfer_manager()->TransferLiteralFromDevice( local
172 if (result_literal.ok()) {
174 << result_literal.ValueOrDie()->ToString();
177 << result_literal.status().ToString();
179 return result_literal;
hlo_evaluator_test.cc 1416 std::unique_ptr<Literal> result_literal = local
    [all...]
  /external/tensorflow/tensorflow/compiler/xla/
reference_util.cc 579 std::unique_ptr<Literal> result_literal = local
582 CHECK_EQ(ShapeUtil::Rank(result_literal->shape()), 4);
584 MakeUnique<Array4D<float>>(result_literal->shape().dimensions(0),
585 result_literal->shape().dimensions(1),
586 result_literal->shape().dimensions(2),
587 result_literal->shape().dimensions(3));
590 *value = result_literal->Get<float>(indices);
literal_util.cc 820 auto result_literal = MakeUnique<Literal>(result_shape); local
825 result_literal->EachCell<float>(
831 result_literal->Set<float>(indices, value);
833 return result_literal;
835 result_literal->EachCell<complex64>(
841 result_literal->Set<complex64>(indices, value);
843 return result_literal;
845 result_literal->EachCell<int32>(
851 result_literal->Set<int32>(indices, value);
853 return result_literal;
1291 auto result_literal = MakeUnique<Literal>(ShapeUtil::ChangeElementType( local
1307 auto result_literal = MakeUnique<Literal>( local
    [all...]
  /external/tensorflow/tensorflow/compiler/xla/tests/
local_client_execute_test.cc 219 std::unique_ptr<Literal> result_literal = ShapedBufferToLiteral(*result); local
221 {{1.0f, 2.0f}, {3.0f, 4.0f}}, LiteralView::Create(*result_literal, {0}));
224 LiteralView::Create(*result_literal, {1}));
226 {{1.0f, 2.0f}, {3.0f, 4.0f}}, LiteralView::Create(*result_literal, {2}));
248 std::unique_ptr<Literal> result_literal = ShapedBufferToLiteral(*result); local
250 {{1.0f, 2.0f}, {3.0f, 4.0f}}, LiteralView::Create(*result_literal, {1}));
253 LiteralView::Create(*result_literal, {0, 0}));
256 LiteralView::Create(*result_literal, {0, 1}));
259 LiteralView::Create(*result_literal, {0, 2}));
283 std::unique_ptr<Literal> result_literal = ShapedBufferToLiteral(*result) local
329 std::unique_ptr<Literal> result_literal = ShapedBufferToLiteral(*result); local
371 std::unique_ptr<Literal> result_literal = ShapedBufferToLiteral(*result); local
459 std::unique_ptr<Literal> result_literal = ShapedBufferToLiteral(*result); local
516 std::unique_ptr<Literal> result_literal = ShapedBufferToLiteral(*result); local
561 std::unique_ptr<Literal> result_literal = ShapedBufferToLiteral(*result); local
    [all...]

Completed in 490 milliseconds