Home
last modified time | relevance | path

Searched refs:bias_add_op (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/lite/toco/graph_transformations/
Didentify_dilated_conv.cc114 Operator* bias_add_op = !has_bias_before_bts ? final_op : next_op; in ResolveDilatedConv() local
115 if (bias_add_op->type != OperatorType::kAdd) { in ResolveDilatedConv()
119 CHECK_EQ(bias_add_op->inputs.size(), 2); in ResolveDilatedConv()
120 CHECK_EQ(bias_add_op->outputs.size(), 1); in ResolveDilatedConv()
130 bias_add_op->outputs[0] = final_op->outputs[0]; in ResolveDilatedConv()
132 bias_add_op->inputs[0] = post_conv_op->outputs[0]; in ResolveDilatedConv()
137 bias_add_op->inputs[0] = conv_op->outputs[0]; in ResolveDilatedConv()
/external/tensorflow/tensorflow/contrib/quantize/python/
Dquantize.py147 layer_match.bias_add_op,
148 input_to_ops_map.ConsumerOperations(layer_match.bias_add_op),
158 quantized_ops.add(layer_match.bias_add_op)
492 bias_add_op = match_result.get_op(bias_add_pattern)
493 if bias_add_op is None:
494 bias_add_op = match_result.get_op(folded_bias_add_pattern)
502 post_activation_bypass_op, bias_add_op))
515 bias_add_op = match_result.get_op(bias_add_pattern)
516 if bias_add_op is None:
517 bias_add_op = match_result.get_op(folded_bias_add_pattern)
[all …]
Dquantize_test.py243 bias_add_op = graph.get_operation_by_name('test/BiasAdd')
246 [op.type for op in bias_add_op.outputs[0].consumers()])
/external/tensorflow/tensorflow/python/tools/
Doptimize_for_inference_lib.py352 bias_add_op = node_def_pb2.NodeDef()
353 bias_add_op.op = "BiasAdd"
354 bias_add_op.name = node.name
355 bias_add_op.attr["T"].CopyFrom(conv_op.attr["T"])
356 bias_add_op.attr["data_format"].CopyFrom(conv_op.attr["data_format"])
357 bias_add_op.input.extend([new_conv_op.name, offset_op.name])
358 new_ops.extend([scaled_weights_op, new_conv_op, offset_op, bias_add_op])
/external/tensorflow/tensorflow/tools/graph_transforms/
Dquantize_nodes_test.cc358 Output bias_add_op = in TestQuantizeBiasAdd() local
889 Output bias_add_op = in TestConvertFakeQuantsToRequantize() local
903 FakeQuantWithMinMaxVars(root.WithOpName("fake_quant_op"), bias_add_op, in TestConvertFakeQuantsToRequantize()
1041 Output bias_add_op = in TestConvertFakeQuantsEndToEnd() local
1055 FakeQuantWithMinMaxVars(root.WithOpName("fake_quant_op"), bias_add_op, in TestConvertFakeQuantsEndToEnd()
1090 Output bias_add_op = in TestHoistFakeQuants() local
1093 Output relu_op = Relu(root.WithOpName("relu_op"), bias_add_op); in TestHoistFakeQuants()