Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 2 of 2 for requested_device (0.16 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_;
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Sat Oct 12 05:11:17 UTC 2024
    - 23.4K 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>(
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Sat Oct 12 16:27:48 UTC 2024
    - 102.3K bytes
    - Viewed (0)
Back to top