| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | fused_kernel_matcher.cc | 80 auto bias_add = dyn_cast_or_null<BiasAddOp>(use.getOwner()); in GetBiasAdd() local
104 virtual bool AreFuseCompatible(SrcOpT contraction_op, BiasAddOp bias_add, in AreFuseCompatible()
111 virtual bool IsDeviceCompatible(SrcOpT contraction_op, BiasAddOp bias_add, in IsDeviceCompatible()
135 BiasAddOp bias_add = GetBiasAdd(contraction.getResult()); in matchAndRewrite() local
245 bool AreFuseCompatible(Conv2DOp conv, BiasAddOp bias_add, in AreFuseCompatible()
266 bool IsDeviceCompatible(Conv2DOp conv, BiasAddOp bias_add, in IsDeviceCompatible()
290 bool AreFuseCompatible(MatMulOp matmul, BiasAddOp bias_add, in AreFuseCompatible()
303 bool IsDeviceCompatible(MatMulOp matmul, BiasAddOp bias_add, in IsDeviceCompatible()
|
| /external/tensorflow/tensorflow/core/transforms/remapper/ |
| D | remapping_helper.h | 37 Operation* bias_add; member
42 Operation* bias_add; member
48 Operation* bias_add; member
54 Operation* bias_add; member
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | remapper_test.cc | 664 auto bias_add = ops::BiasAdd(s.WithOpName("bias_add"), conv, bias); in RunTest() local
669 auto bias_add = ops::BiasAdd(s.WithOpName("bias_add"), conv, bias); in RunTest() local
774 auto bias_add = ops::BiasAdd(s.WithOpName("bias_add"), conv, bias); in RunTest() local
797 auto bias_add = ops::BiasAdd(s.WithOpName("bias_add"), conv, bias); in RunTest() local
930 auto bias_add = ops::BiasAdd(s.WithOpName("bias_add"), squeeze, bias); in RunTest() local
939 auto bias_add = ops::BiasAdd(s.WithOpName("bias_add"), squeeze, bias); in RunTest() local
1293 auto bias_add = ops::BiasAdd(s.WithOpName("bias_add"), conv, bias); in RunTest() local
1434 auto bias_add = ops::BiasAdd(s.WithOpName("bias_add"), matmul, bias); in RunTest() local
1530 auto bias_add = ops::BiasAdd(s.WithOpName("bias_add"), conv, bias); in TEST_F() local
1611 auto bias_add = ops::BiasAdd(s.WithOpName("bias_add"), matmul, bias); in RunTest() local
[all …]
|
| D | mkl_remapper_test.cc | 73 auto bias_add = ops::BiasAdd(s.WithOpName("bias_add"), conv, bias, in FuseConv2DWithBiasAndAddNOrAdd() local
467 Output bias_add; in TEST_F() local
559 Output bias_add; in RelpaceAddWithBiasAddDepthwiseConv2D() local
668 auto bias_add = ops::BiasAdd(s.WithOpName("bias_add"), matmul, bias); in RunTest() local
|
| D | remapper.cc | 167 int bias_add = kMissingIndex; member
182 int bias_add = kMissingIndex; member
195 int bias_add = kMissingIndex; member
240 int bias_add = kMissingIndex; member
261 int bias_add = kMissingIndex; member
2215 const NodeDef& bias_add = graph->node(matched.bias_add); in AddFusedContractionNode() local
2260 const NodeDef& bias_add = graph->node(matched.bias_add); in AddFusedContractionNode() local
2315 const NodeDef& bias_add = graph->node(matched.bias_add); in AddFusedConvNode() local
2448 const NodeDef& bias_add = graph->node(matched.bias_add); in AddFusedContractionNode() local
2565 const NodeDef& bias_add = graph->node(matched.bias_add); in AddFusedContractionNode() local
|
| D | generic_layout_optimizer_test.cc | 502 auto bias_add = ops::BiasAdd( in TEST_F() local
|
| /external/tensorflow/tensorflow/python/kernel_tests/nn_ops/ |
| D | bias_op_base.py | 156 def bias_add(input_tensor, bias_tensor): function
|
| /external/tensorflow/tensorflow/core/grappler/utils/ |
| D | pattern_utils_test.cc | 43 auto bias_add = ops::BiasAdd(s.WithOpName("bias_add"), matmul, bias); in GetMatMulBiasAddGeluGraph() local
|
| /external/tensorflow/tensorflow/tools/graph_transforms/ |
| D | quantize_nodes_test.cc | 831 Output bias_add = in TestInputRange() local
856 Output bias_add = in TestFallbackRange() local
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | nn_ops.py | 3512 def bias_add(value, bias, data_format=None, name=None): function
|
| /external/tensorflow/tensorflow/python/keras/ |
| D | backend.py | 5980 def bias_add(x, bias, data_format=None): function
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | mkl_layout_pass.cc | 861 string bias_add; member
|