Searched refs:RegisterStatistics (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/python/profiler/internal/ |
| D | flops_registry.py | 71 @ops.RegisterStatistics("Reciprocal", "flops")
77 @ops.RegisterStatistics("Square", "flops")
83 @ops.RegisterStatistics("Rsqrt", "flops")
90 @ops.RegisterStatistics("Log", "flops")
96 @ops.RegisterStatistics("Neg", "flops")
102 @ops.RegisterStatistics("AssignSub", "flops")
108 @ops.RegisterStatistics("AssignAdd", "flops")
114 @ops.RegisterStatistics("L2Loss", "flops")
124 @ops.RegisterStatistics("Softmax", "flops")
147 @ops.RegisterStatistics("Add", "flops")
[all …]
|
| /external/tensorflow/tensorflow/core/profiler/g3doc/ |
| D | profile_model_architecture.md | 48 * It must have `RegisterStatistics('flops')` defined in TensorFlow. tfprof
52 * It must have known "shape" information for RegisterStatistics('flops') to
|
| D | options.md | 95 … this number of float operations. Only available if an node has op.RegisterStatistics() defined an…
|
| /external/tensorflow/tensorflow/core/profiler/ |
| D | tfprof_log.proto | 31 // op has RegisterStatistics defined. Currently, Conv2D, MatMul, etc, are
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | nn_ops.py | 3926 @ops.RegisterStatistics("Conv2D", "flops")
3945 @ops.RegisterStatistics("DepthwiseConv2dNative", "flops")
3961 @ops.RegisterStatistics("BiasAdd", "flops")
4521 @ops.RegisterStatistics("Dilation2D", "flops")
|
| D | math_ops.py | 2688 @ops.RegisterStatistics("MatMul", "flops")
|
| /external/tensorflow/tensorflow/python/framework/ |
| D | ops.py | 2832 class RegisterStatistics(object): class
|
| D | ops_test.py | 2659 @ops.RegisterStatistics("a", "flops")
|