Home
last modified time | relevance | path

Searched refs:bias_add (Results 1 – 25 of 54) sorted by relevance

123

/external/tensorflow/tensorflow/core/grappler/optimizers/
Dremapper.cc74 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 …]
Dremapper_test.cc121 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/
Dbiasadd_matmul_test.py75 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")
Ddynamic_input_shapes_test.py59 x = nn.bias_add(x, bias1)
70 x = nn.bias_add(x, bias2)
Dint32_test.py50 x = nn.bias_add(x, b)
Dvgg_block_nchw_test.py61 t = nn.bias_add(conv, b, data_format="NCHW", name="biasAdd")
Dvgg_block_test.py55 t = nn.bias_add(conv, b, name="biasAdd")
/external/tensorflow/tensorflow/python/kernel_tests/
Dbias_op_test.py51 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/
Drnn_cell.py139 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/
Dcore_layers.py195 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)
Drnn_cells.py149 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/
Dfused_conv2d_bias_activation_benchmark.py64 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/
Drecurrent.py1243 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 …]
Drecurrent_v2.py364 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)
Dconvolutional.py205 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(
Dkernelized.py206 outputs = nn.bias_add(outputs, self.bias)
Dlocal.py254 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/
Drnn_cell.py276 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 …]
Dcore_rnn_cell.py121 res = nn_ops.bias_add(res, self._biases)
188 return nn_ops.bias_add(res, biases)
/external/tensorflow/tensorflow/contrib/model_pruning/examples/cifar10/
Dcifar10_pruning.py195 pre_activation = tf.nn.bias_add(conv, biases)
217 pre_activation = tf.nn.bias_add(conv, biases)
/external/tensorflow/tensorflow/core/kernels/
Dconv_ops_fused_impl.h325 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/
Dlayers.py841 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)
Dfeature_column_ops.py426 predictions = nn_ops.bias_add(predictions_no_bias, bias)
540 predictions = nn_ops.bias_add(predictions_no_bias, bias)
/external/tensorflow/tensorflow/python/ops/
Drnn_cell_impl.py466 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/
Dmkl_layout_pass.cc248 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 …]

123