public class OpProfiler extends Object
Modifier and Type | Class and Description |
---|---|
static interface |
OpProfiler.OpProfilerListener |
static class |
OpProfiler.PenaltyCause |
Modifier and Type | Method and Description |
---|---|
void |
addListener(OpProfiler.OpProfilerListener listener)
dd the listener to the profiler
|
void |
clearListener(OpProfiler.OpProfilerListener listener)
Clear the listener from the profiler
|
static OpProfiler |
getInstance() |
long |
getInvocationsCount() |
StackAggregator |
getMixedOrderAggregator()
Dev-time method.
|
protected String |
getOpClass(Op op)
This method returns op class name
|
StackAggregator |
getScalarAggregator() |
protected OpProfiler.PenaltyCause[] |
joinDistinct(OpProfiler.PenaltyCause[] a,
OpProfiler.PenaltyCause[] b) |
void |
printOutDashboard()
This method prints out dashboard state
|
void |
processBlasCall(boolean isGemm,
INDArray... operands) |
void |
processBlasCall(String blasOpName)
Deprecated.
|
void |
processMemoryAccess() |
void |
processOpCall(Op op)
This method tracks op calls
|
void |
processOpCall(Op op,
DataBuffer... tadBuffers) |
OpProfiler.PenaltyCause[] |
processOperands(INDArray... operands)
This method checks for something somewhere
|
OpProfiler.PenaltyCause[] |
processOperands(INDArray x,
INDArray y) |
OpProfiler.PenaltyCause[] |
processOperands(INDArray x,
INDArray y,
INDArray z) |
String |
processOrders(INDArray... operands) |
void |
processScalarCall()
This method tracks INDArray.putScalar()/getScalar() calls
|
void |
processStackCall(Op op,
long timeStart)
This method builds
|
OpProfiler.PenaltyCause[] |
processTADOperands(DataBuffer... tadBuffers) |
void |
reset()
This method resets all counters
|
void |
timeBlasCall() |
void |
timeOpCall(Op op,
long startTime) |
protected void |
updatePairs(String opName,
String opClass) |
public void clearListener(OpProfiler.OpProfilerListener listener)
listener
- the listener to clearpublic void addListener(OpProfiler.OpProfilerListener listener)
listener
- the listener to addpublic void reset()
public static OpProfiler getInstance()
protected String getOpClass(Op op)
op
- public void processScalarCall()
public void processOpCall(Op op)
op
- public void processOpCall(Op op, DataBuffer... tadBuffers)
op
- tadBuffers
- public StackAggregator getMixedOrderAggregator()
public StackAggregator getScalarAggregator()
public void timeOpCall(Op op, long startTime)
@Deprecated public void processBlasCall(String blasOpName)
public void timeBlasCall()
public void printOutDashboard()
public long getInvocationsCount()
public void processStackCall(Op op, long timeStart)
op
- public void processBlasCall(boolean isGemm, INDArray... operands)
public OpProfiler.PenaltyCause[] processOperands(INDArray x, INDArray y)
public OpProfiler.PenaltyCause[] processTADOperands(DataBuffer... tadBuffers)
public OpProfiler.PenaltyCause[] processOperands(INDArray x, INDArray y, INDArray z)
protected OpProfiler.PenaltyCause[] joinDistinct(OpProfiler.PenaltyCause[] a, OpProfiler.PenaltyCause[] b)
public OpProfiler.PenaltyCause[] processOperands(INDArray... operands)
operands
- public void processMemoryAccess()
Copyright © 2017. All rights reserved.