Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 3 of 3 for set_device (0.1 sec)

  1. tensorflow/c/experimental/stream_executor/stream_executor_internal.h

          event_handle_ = nullptr;
        }
      }
    
      SP_Event Handle() { return event_handle_; }
    
     private:
      SP_Device* device_;
      SP_StreamExecutor* stream_executor_;
      SP_Event event_handle_;
    };
    
    class CStream : public StreamCommon {
     public:
      CStream(SP_Device* device, SP_StreamExecutor* stream_executor,
              StreamExecutor* executor)
          : StreamCommon(executor),
            device_(device),
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri Jun 14 07:39:19 UTC 2024
    - 8K bytes
    - Viewed (0)
  2. tensorflow/c/experimental/stream_executor/stream_executor.cc

                              SP_DEVICE_MEMORY_BASE_STRUCT_SIZE);
      // All other fields could theoretically be zero/null.
      return absl::OkStatus();
    }
    
    absl::Status ValidateSPDevice(const SP_Device& device) {
      TF_VALIDATE_STRUCT_SIZE(SP_Device, device, SP_DEVICE_STRUCT_SIZE);
      // All other fields could theoretically be zero/null.
      return absl::OkStatus();
    }
    
    absl::Status ValidateSPDeviceFns(const SP_DeviceFns& device_fns) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri Jun 14 07:39:19 UTC 2024
    - 27.1K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/tensorflow/ir/tf_generated_ops.td

        }
      }];
    }
    
    def TF_RecvOp : TF_Op<"Recv", [TF_RecvSideEffect]> {
      let summary = "Receives the named tensor from send_device on recv_device.";
    
      let arguments = (ins
        StrAttr:$tensor_name,
        StrAttr:$send_device,
        I64Attr:$send_device_incarnation,
        StrAttr:$recv_device,
        DefaultValuedOptionalAttr<BoolAttr, "false">:$client_terminated
      );
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Jun 11 23:24:08 UTC 2024
    - 793K bytes
    - Viewed (0)
Back to top