Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 5 of 5 for their (0.29 sec)

  1. tensorflow/c/eager/parallel_device/parallel_device.cc

            //
            // TODO(allenl): There may be smarter ways to do this copy in some
            // cases, i.e. with a collective broadcast. We'll need to be careful
            // about things that are taken as inputs on the host or on their
            // existing device (for multi-device functions).
            std::unique_ptr<ParallelTensor> parallel_tensor(
                parallel_device.CopyToParallelDevice(
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Wed Mar 29 22:05:31 GMT 2023
    - 18.3K bytes
    - Viewed (0)
  2. tensorflow/c/eager/custom_device_testutil.cc

      if (dev->strict_scope_placement && *requested_placement == '\0') {
        TF_SetStatus(s, TF_INTERNAL,
                     "Ops must be placed on the device explicitly, or their inputs "
                     "first copied to other devices.");
        return;
      }
      TFE_Context* context = TFE_OpGetContext(original_op, s);
      if (TF_GetCode(s) != TF_OK) return;
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Wed Mar 03 20:47:31 GMT 2021
    - 8.3K bytes
    - Viewed (0)
  3. tensorflow/c/c_api_experimental.cc

      // CPU tensor to GPU).
      // Setting a larger thread pool does not help with the Swift caller, as we use
      // a different TFE context for each thread of execution (for running graph
      // functions, and their send/recvs corountines).
      config.set_inter_op_parallelism_threads(1);
    
      TF_Buffer* ret = TF_NewBuffer();
      TF_CHECK_OK(MessageToBuffer(config, ret));
      return ret;
    }
    
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Mon Apr 15 03:35:10 GMT 2024
    - 29.4K bytes
    - Viewed (0)
  4. tensorflow/c/experimental/filesystem/plugins/gcs/ram_file_block_cache_test.cc

              std::vector<char> x(block_size, 'x');
              EXPECT_EQ(out, x);
            }));
      }
      // The `threads` destructor blocks until the threads can be joined, once their
      // respective reads finish (which happens once they are all concurrently being
      // executed, or 10 seconds have passed).
    }
    
    TEST(RamFileBlockCacheTest, CoalesceConcurrentReads) {
    C++
    - Registered: Tue Apr 23 12:39:09 GMT 2024
    - Last Modified: Fri Oct 15 03:16:57 GMT 2021
    - 23.2K bytes
    - Viewed (0)
  5. tensorflow/c/c_api_function_test.cc

      ASSERT_EQ(TF_OK, TF_GetCode(s_)) << TF_Message(s_);
      TF_GraphCopyFunction(host_graph_, func2, grad_func, s_);
      ASSERT_EQ(TF_OK, TF_GetCode(s_)) << TF_Message(s_);
    
      // Verify that functions and their gradients are in host graph's GraphDef
      GraphDef gdef;
      GetGraphDef(host_graph_, &gdef);
      std::vector<std::pair<string, string>> grads = GetGradDefs(gdef);
      ASSERT_EQ(2, grads.size());
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Thu Jul 20 22:08:54 GMT 2023
    - 63.6K bytes
    - Viewed (6)
Back to top