Caffe2 - Python API
A deep learning, cross platform ML framework
Public Member Functions | List of all members
caffe2.python.device_checker.DeviceChecker Class Reference
Inheritance diagram for caffe2.python.device_checker.DeviceChecker:

Public Member Functions

def __init__ (self, threshold, device_options)
def CheckSimple (self, op, inputs, outputs_to_check, input_device_options=None)
def CheckNet (self, net, inputs=None, blobs_to_check=None, ignore=None)

Detailed Description

A device checker in Python to check consistency across multiple devices.

This is not the most efficient way to check devices, as the Python interface
will involve a lot of copies back and forth operations. Use at your own risk.

Definition at line 10 of file

Member Function Documentation

def caffe2.python.device_checker.DeviceChecker.CheckNet (   self,
  inputs = None,
  blobs_to_check = None,
  ignore = None 
Checks a network by inspecting all of its intermediate results, and
see if things match.

Definition at line 76 of file

def caffe2.python.device_checker.DeviceChecker.CheckSimple (   self,
  input_device_options = None 
Checks the operator with different device implementations.

  op: the operator to be checked.
  inputs: the input data in numpy arrays.
  outputs_to_check: the outputs to check between devices.
  input_device_options: a mapping from input name to a device to use
    (instead of self._device_options)
  boolean: True if it passes, False if it does not pass.

Definition at line 22 of file

The documentation for this class was generated from the following file: