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 24 of file device_checker.py.

Member Function Documentation

def caffe2.python.device_checker.DeviceChecker.CheckNet (   self,
  net,
  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 87 of file device_checker.py.

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

Inputs:
  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)
Outputs:
  boolean: True if it passes, False if it does not pass.

Definition at line 36 of file device_checker.py.


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