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

---

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

• caffe2/python/device_checker.py