Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 1 of 1 for CreateInputTensor (4.86 sec)

  1. tensorflow/compiler/jit/tests/device_compiler_test_helper.cc

        std::unique_ptr<Session> session(NewSession(options));
        TF_RETURN_IF_ERROR(session->Create(graph));
        RunOptions run_options;
    
        Tensor input_a = CreateInputTensor(shape, 0);
        Tensor input_b = CreateInputTensor(shape, shape.num_elements());
        Tensor input_c = CreateInputTensor(shape, 2 * shape.num_elements());
        TF_RETURN_IF_ERROR(session->Run(
            run_options,
            {std::make_pair("a", input_a), std::make_pair("b", input_b),
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri Feb 09 08:24:16 UTC 2024
    - 6.2K bytes
    - Viewed (0)
Back to top