Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 2 of 2 for requested_device (0.22 sec)

  1. tensorflow/c/eager/c_api_distributed_test.cc

        // the given node name (error_node_) and requested device (error_device_).
        for (const auto node : graph->get()->nodes()) {
          if (node->name().find(error_node_) != string::npos &&
              node->requested_device() == error_device_) {
            return tensorflow::errors::Internal("Injected graph pass error.");
          }
        }
        return absl::OkStatus();
      }
    
     private:
      const string error_node_;
    C++
    - Registered: Tue Apr 23 12:39:09 GMT 2024
    - Last Modified: Thu Feb 15 09:49:45 GMT 2024
    - 23.5K bytes
    - Viewed (0)
  2. tensorflow/c/c_api.cc

    }
    
    const char* TF_OperationOpType(TF_Operation* oper) {
      return oper->node.type_string().c_str();
    }
    
    const char* TF_OperationDevice(TF_Operation* oper) {
      return oper->node.requested_device().c_str();
    }
    
    int TF_OperationNumOutputs(TF_Operation* oper) {
      return oper->node.num_outputs();
    }
    
    TF_DataType TF_OperationOutputType(TF_Output oper_out) {
      return static_cast<TF_DataType>(
    C++
    - Registered: Tue Apr 23 12:39:09 GMT 2024
    - Last Modified: Mon Apr 15 03:35:10 GMT 2024
    - 102.3K bytes
    - Viewed (0)
Back to top