/external/tensorflow/tensorflow/core/grappler/optimizers/ |
D | remapper.cc | 74 Conv2DWithBiasAdd(const NodeDef* conv2d, const NodeDef* bias_add) in Conv2DWithBiasAdd() 75 : conv2d(conv2d), bias_add(bias_add) {} in Conv2DWithBiasAdd() 78 const NodeDef* bias_add = nullptr; member 84 Conv2DWithBiasAddAndRelu(const NodeDef* conv2d, const NodeDef* bias_add, in Conv2DWithBiasAddAndRelu() 86 : conv2d(conv2d), bias_add(bias_add), relu(relu) {} in Conv2DWithBiasAddAndRelu() 89 const NodeDef* bias_add = nullptr; member 97 const NodeDef* bias_add) in Conv2DWithSqueezeAndBiasAdd() 98 : conv2d(conv2d), squeeze(squeeze), bias_add(bias_add) {} in Conv2DWithSqueezeAndBiasAdd() 102 const NodeDef* bias_add = nullptr; member 229 bool FindConv2DWithBias(const RemapperContext& ctx, const NodeDef* bias_add, in FindConv2DWithBias() argument [all …]
|
D | remapper_test.cc | 121 auto bias_add = ops::BiasAdd(s.WithOpName("bias_add"), conv, bias); in TEST_F() local 122 auto fetch = ops::Identity(s.WithOpName("fetch"), bias_add); in TEST_F() 184 auto bias_add = ops::BiasAdd(s.WithOpName("bias_add"), conv, bias); in TEST_F() local 185 auto relu = ops::Relu(s.WithOpName("relu"), bias_add); in TEST_F() 410 auto bias_add = ops::BiasAdd(s.WithOpName("bias_add"), squeeze, bias); in TEST_F() local 411 auto fetch = ops::Identity(s.WithOpName("fetch"), bias_add); in TEST_F()
|
/external/tensorflow/tensorflow/python/compiler/tensorrt/test/ |
D | biasadd_matmul_test.py | 75 x5 = nn.bias_add(x5, b) 80 x6 = nn.bias_add(x6, b, data_format="NHWC") 85 x7 = nn.bias_add(x7, b, data_format="NHWC") 90 x8 = nn.bias_add(x8, b, data_format="NHWC") 95 x9 = nn.bias_add(x9, b, data_format="NCHW") 100 x10 = nn.bias_add(x10, b, data_format="NCHW") 105 x11 = nn.bias_add(x11, b, data_format="NCHW")
|
D | dynamic_input_shapes_test.py | 59 x = nn.bias_add(x, bias1) 70 x = nn.bias_add(x, bias2)
|
D | int32_test.py | 50 x = nn.bias_add(x, b)
|
D | vgg_block_nchw_test.py | 61 t = nn.bias_add(conv, b, data_format="NCHW", name="biasAdd")
|
D | vgg_block_test.py | 55 t = nn.bias_add(conv, b, name="biasAdd")
|
/external/tensorflow/tensorflow/python/kernel_tests/ |
D | bias_op_test.py | 51 tf_val = nn_ops.bias_add(np_inputs, np_bias).eval() 79 tf_val = nn_ops.bias_add(np_inputs, np_bias, data_format="NCHW").eval() 94 nn_ops.bias_add([1, 2], [1]) 99 nn_ops.bias_add( 108 nn_ops.bias_add( 148 output_tensor = nn_ops.bias_add( 171 output_tensor = nn_ops.bias_add(
|
/external/tensorflow/tensorflow/lite/experimental/examples/lstm/ |
D | rnn_cell.py | 139 gate_inputs = nn_ops.bias_add(gate_inputs, self._bias) 451 i = nn_ops.bias_add( 457 f = nn_ops.bias_add( 463 o = nn_ops.bias_add( 469 j = nn_ops.bias_add(
|
/external/tensorflow/tensorflow/contrib/model_pruning/python/layers/ |
D | core_layers.py | 195 outputs = nn.bias_add(outputs, self.bias, data_format='NCHW') 205 outputs_4d = nn.bias_add(outputs_4d, self.bias, data_format='NCHW') 208 outputs = nn.bias_add(outputs, self.bias, data_format='NHWC') 466 outputs = nn.bias_add(outputs, self.bias)
|
D | rnn_cells.py | 149 gate_inputs = nn_ops.bias_add(gate_inputs, self._bias) 314 lstm_matrix = nn_ops.bias_add(lstm_matrix, self._bias)
|
/external/tensorflow/tensorflow/contrib/fused_conv/python/ops/ |
D | fused_conv2d_bias_activation_benchmark.py | 64 bias_out = nn_ops.bias_add(conv2d_out, bias, data_format=data_format) 71 bias_out = nn_ops.bias_add(conv2d_out, bias, data_format=data_format)
|
/external/tensorflow/tensorflow/python/keras/layers/ |
D | recurrent.py | 1243 h = K.bias_add(h, self.bias) 1675 x_z = K.bias_add(x_z, input_bias[:self.units]) 1676 x_r = K.bias_add(x_r, input_bias[self.units: self.units * 2]) 1677 x_h = K.bias_add(x_h, input_bias[self.units * 2:]) 1692 recurrent_z = K.bias_add(recurrent_z, recurrent_bias[:self.units]) 1693 recurrent_r = K.bias_add(recurrent_r, 1703 recurrent_h = K.bias_add(recurrent_h, recurrent_bias[self.units * 2:]) 1718 matrix_x = K.bias_add(matrix_x, input_bias) 1731 matrix_inner = K.bias_add(matrix_inner, recurrent_bias) 2242 x_i = K.bias_add(x_i, b_i) [all …]
|
D | recurrent_v2.py | 364 matrix_x = K.bias_add(matrix_x, input_bias) 370 matrix_inner = K.bias_add(matrix_inner, recurrent_bias) 763 z = K.bias_add(z, bias)
|
D | convolutional.py | 205 outputs = nn.bias_add(outputs, self.bias, data_format='NCHW') 207 outputs = nn.bias_add(outputs, self.bias, data_format='NHWC') 829 outputs = nn.bias_add( 1111 outputs = nn.bias_add( 1507 outputs = nn.bias_add( 1661 outputs = nn.bias_add( 1817 outputs = backend.bias_add(
|
D | kernelized.py | 206 outputs = nn.bias_add(outputs, self.bias)
|
D | local.py | 254 output = K.bias_add(output, self.bias, data_format=self.data_format) 542 output = K.bias_add(output, self.bias, data_format=self.data_format)
|
/external/tensorflow/tensorflow/contrib/rnn/python/ops/ |
D | rnn_cell.py | 276 lstm_matrix = nn_ops.bias_add(lstm_matrix, b) 444 lstm_matrix = nn_ops.bias_add(math_ops.matmul(cell_inputs, concat_w), b) 764 lstm_matrix_freq = nn_ops.bias_add( 780 lstm_matrix_time = nn_ops.bias_add( 1435 out = nn_ops.bias_add(out, bias) 1577 m_matrix = nn_ops.bias_add(m_matrix, self.bias) 2446 i = nn_ops.bias_add(array_ops.concat(i_parts, axis=1), bi) 2447 j = nn_ops.bias_add(array_ops.concat(j_parts, axis=1), bj) 2448 f = nn_ops.bias_add(array_ops.concat(f_parts, axis=1), bf) 2449 o = nn_ops.bias_add(array_ops.concat(o_parts, axis=1), bo) [all …]
|
D | core_rnn_cell.py | 121 res = nn_ops.bias_add(res, self._biases) 188 return nn_ops.bias_add(res, biases)
|
/external/tensorflow/tensorflow/contrib/model_pruning/examples/cifar10/ |
D | cifar10_pruning.py | 195 pre_activation = tf.nn.bias_add(conv, biases) 217 pre_activation = tf.nn.bias_add(conv, biases)
|
/external/tensorflow/tensorflow/core/kernels/ |
D | conv_ops_fused_impl.h | 325 BiasAddArgs bias_add; 334 OP_REQUIRES_OK(context, InitBiasAddArgs(context, &bias_add)); 335 conv2d(WithBiasAdd<T>(bias_add.bias_add_data), context, input, filter, 340 OP_REQUIRES_OK(context, InitBiasAddArgs(context, &bias_add)); 341 conv2d(WithBiasAddAndRelu<T>(bias_add.bias_add_data), context, input,
|
/external/tensorflow/tensorflow/contrib/layers/python/layers/ |
D | layers.py | 841 def bias_add(inputs, function 909 outputs = nn.bias_add(inputs, biases, data_format=data_format) 1301 outputs = nn.bias_add(outputs, biases) 2078 norm_pool = nn.bias_add(norm_pool, self.beta, data_format='NCHW') 2083 norm_pool = nn.bias_add(norm_pool, self.beta, data_format='NCHW') 2086 norm_pool = nn.bias_add(norm_pool, self.beta, data_format='NCHW') 2089 norm_pool = nn.bias_add(norm_pool, self.beta, data_format='NHWC') 2844 outputs = nn.bias_add(outputs, biases, data_format=data_format) 3303 y = nn.bias_add(y, b)
|
D | feature_column_ops.py | 426 predictions = nn_ops.bias_add(predictions_no_bias, bias) 540 predictions = nn_ops.bias_add(predictions_no_bias, bias)
|
/external/tensorflow/tensorflow/python/ops/ |
D | rnn_cell_impl.py | 466 gate_inputs = nn_ops.bias_add(gate_inputs, self._bias) 582 gate_inputs = nn_ops.bias_add(gate_inputs, self._gate_bias) 591 candidate = nn_ops.bias_add(candidate, self._candidate_bias) 768 gate_inputs = nn_ops.bias_add(gate_inputs, self._bias) 1020 lstm_matrix = nn_ops.bias_add(lstm_matrix, self._bias)
|
/external/tensorflow/tensorflow/core/graph/ |
D | mkl_layout_pass.cc | 248 csinfo_.bias_add = "BiasAdd"; in MklLayoutRewritePass() 539 minfo_.push_back({csinfo_.conv2d, csinfo_.bias_add, in MklLayoutRewritePass() 682 string bias_add; member 870 if (m->type_string() == csinfo_.bias_add) { in GetConv2DOrBiasAdd() 879 e->dst()->type_string() == csinfo_.bias_add && in GetConv2DOrBiasAdd() 2667 CHECK_EQ(((m->type_string() == csinfo_.bias_add && in MergeConv2DWithBiasAdd() 2669 ((n->type_string() == csinfo_.bias_add && in MergeConv2DWithBiasAdd() 2675 Node* pred = m->type_string() == csinfo_.bias_add ? n : m; in MergeConv2DWithBiasAdd() 2676 Node* succ = m->type_string() == csinfo_.bias_add ? m : n; in MergeConv2DWithBiasAdd() 3117 if (((m->type_string() == csinfo_.bias_add && in MergeNode() [all …]
|