| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | fused_kernel_matcher.cc | 77 auto bias_add = dyn_cast_or_null<BiasAddOp>(use.getOwner()); in GetBiasAdd() local
101 virtual bool AreFuseCompatible(SrcOpT contraction_op, BiasAddOp bias_add, in AreFuseCompatible()
115 BiasAddOp bias_add = GetBiasAdd(contraction.getResult()); in matchAndRewrite() local
193 bool AreFuseCompatible(Conv2DOp conv, BiasAddOp bias_add, in AreFuseCompatible()
222 bool AreFuseCompatible(MatMulOp matmul, BiasAddOp bias_add, in AreFuseCompatible()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | remapper.cc | 118 int bias_add = kMissingIndex; member
129 int bias_add = kMissingIndex; member
141 int bias_add = kMissingIndex; member
186 int bias_add = kMissingIndex; member
203 int bias_add = kMissingIndex; member
1067 const NodeDef& bias_add = graph->node(matched.bias_add); in IsCpuCompatibleDataType() local
1111 const NodeDef& bias_add = graph->node(matched.bias_add); in IsCpuCompatibleDataType() local
1164 const NodeDef& bias_add = graph->node(matched.bias_add); in IsCpuCompatibleDataType() local
1292 const NodeDef& bias_add = graph->node(matched.bias_add); in IsCpuCompatibleDataType() local
1349 const NodeDef& bias_add = graph->node(matched.bias_add); in IsCpuCompatibleDataType() local
|
| D | remapper_test.cc | 346 auto bias_add = ops::BiasAdd(s.WithOpName("bias_add"), conv, bias); in TEST_F() local
410 auto bias_add = ops::BiasAdd(s.WithOpName("bias_add"), matmul, bias); in RunTest() local
489 auto bias_add = ops::BiasAdd(s.WithOpName("bias_add"), conv, bias); in TEST_F() local
563 auto bias_add = ops::BiasAdd(s.WithOpName("bias_add"), conv, bias); in TEST_F() local
659 auto bias_add = ops::BiasAdd(s.WithOpName("bias_add"), matmul, bias); in RunTest() local
956 auto bias_add = ops::BiasAdd(s.WithOpName("bias_add"), squeeze, bias); in TEST_F() local
|
| D | mkl_remapper_test.cc | 73 auto bias_add = ops::BiasAdd(s.WithOpName("bias_add"), conv, bias, in FuseConv2DWithBiasAndAddNOrAdd() local
467 auto bias_add = ops::BiasAdd(s.WithOpName("bias_add"), matmul, bias); in TEST_F() local
|
| D | generic_layout_optimizer_test.cc | 532 auto bias_add = ops::BiasAdd( in TEST_F() local
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | bias_op_base.py | 160 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 | 3320 def bias_add(value, bias, data_format=None, name=None): function
|
| /external/tensorflow/tensorflow/python/keras/ |
| D | backend.py | 6053 def bias_add(x, bias, data_format=None): function
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | mkl_layout_pass.cc | 911 string bias_add; member
|