Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 4 of 4 for can (0.15 sec)

  1. tensorflow/c/c_api_experimental.cc

      // TODO(b/113217601): This is needed for EagerContext::runner_ to use a
      // threadpool, so that we avoid the possibility of running the runner_ in the
      // threadpool of GPU event mgr, as that can trigger more callbacks to be
      // scheduled on that same threadpool, causing a deadlock in cases where the
      // caller of event_mgr->ThenExecute() blocks on the completion of the callback
    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)
  2. tensorflow/c/eager/dlpack.cc

      int64_t expected_stride = 1;
      for (int i = ndim - 1; i >= 0; --i) {
        // Empty tensors are always compact regardless of strides.
        if (shape_arr[i] == 0) return true;
        // Note that dimensions with size=1 can have any stride.
        if (shape_arr[i] != 1 && stride_arr[i] != expected_stride) {
          valid = false;
        }
        expected_stride *= shape_arr[i];
      }
      return valid;
    }
    }  // namespace
    
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Thu Feb 15 09:49:45 GMT 2024
    - 12.8K bytes
    - Viewed (0)
  3. tensorflow/c/eager/parallel_device/parallel_device_lib.cc

      CancellationManager* cancellation_manager_ TF_GUARDED_BY(execution_mutex_);
      //   Outputs:
      std::vector<TensorHandlePtr> op_outputs_ TF_GUARDED_BY(execution_mutex_);
      // TF_Status is an incomplete type and so can't be stack allocated. To avoid
      // unnecessary allocations each Execute call, we keep one heap-allocated
      // version for the thread.
      StatusPtr status_ TF_GUARDED_BY(execution_mutex_);
    
      const std::string device_;
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Fri Feb 09 07:47:20 GMT 2024
    - 25.4K bytes
    - Viewed (1)
  4. tensorflow/c/eager/c_api_distributed_test.cc

        }
        return tensorflow::errors::Internal("Graph pass runs for more than once!");
      }
    
     private:
      bool first_call_ = true;
    };
    
    // After the graph pass is registered, it takes effect globally and can affect
    // other test cases. Define a static variable to switch it on and off.
    bool GraphErrorInjectionPass::enabled_ = false;
    
    // Test to ensure that a registered graph optimization pass is only executed
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Thu Feb 15 09:49:45 GMT 2024
    - 23.5K bytes
    - Viewed (0)
Back to top