Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 3 of 3 for Mccall (0.6 sec)

  1. tensorflow/c/eager/c_api.cc

    }
    
    // Note: this function looks up a thread local policy. So it should be called in
    // the appropriate client thread. In particular, in async mode, it may not be
    // safe to call this function from the async EagerExecutor threads.
    extern TFE_ContextDevicePlacementPolicy TFE_ContextGetDevicePlacementPolicy(
        TFE_Context* ctx) {
      return static_cast<TFE_ContextDevicePlacementPolicy>(
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Tue Mar 12 20:00:09 GMT 2024
    - 43.9K bytes
    - Viewed (2)
  2. tensorflow/c/eager/tape.h

          new GradientTape<Gradient, BackwardFunction, TapeTensor>(false));
      AccumulatorCallState& call_state = call_state_.top();
      call_state.backward_tape = tape.get();
      auto pop_backward_tape =
          gtl::MakeCleanup([&call_state] { call_state.backward_tape = nullptr; });
      std::vector<Gradient*> forwardprop_aids;
      std::vector<int64_t> sources;
      std::unordered_set<int64_t> sources_set;
    C
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Tue Apr 02 12:40:29 GMT 2024
    - 47.2K bytes
    - Viewed (1)
  3. tensorflow/c/c_api_experimental.cc

      using tensorflow::AttrBuilder::AttrBuilder;
      // The string buffers to make sure that any `attr_name` we pass into
      // `builder->Set()` will outlive the subsequent
      // `TF_AttrBuilderCheckCanRunOnDevice()` call(s) on the same `builder`.
      std::set<std::string> attr_names;
    };
    
    TF_AttrBuilder* TF_NewAttrBuilder(const char* op_name) {
      return new TF_AttrBuilder(op_name);
    }
    
    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)
Back to top