Searched refs:bias_add_op (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/c/experimental/ops/ |
| D | nn_ops.cc | 75 AbstractOperationPtr bias_add_op(ctx->CreateOperation()); in BiasAdd() local
77 bias_add_op->Reset("BiasAdd", /*raw_device_name=*/nullptr)); in BiasAdd()
78 TF_RETURN_IF_ERROR(MaybeSetOpName(bias_add_op.get(), name)); in BiasAdd()
79 TF_RETURN_IF_ERROR(bias_add_op->AddInput(inputs[0])); // tensor input in BiasAdd()
80 TF_RETURN_IF_ERROR(bias_add_op->AddInput(inputs[1])); // bias in BiasAdd()
83 TF_RETURN_IF_ERROR(bias_add_op->Execute(outputs, &num_retvals)); in BiasAdd()
|
| /external/tensorflow/tensorflow/lite/delegates/hexagon/builders/ |
| D | matmul_builder.cc | 102 auto* bias_add_op = graph_builder->AddNode(matmul_op->GetTFLiteNodeID()); in AddFullyConnectedHelper() local
103 bias_add_op->SetOpType(OP_QuantizedBiasAdd_32p32to32); in AddFullyConnectedHelper()
104 bias_add_op->AddInput(matmul_out); in AddFullyConnectedHelper()
105 bias_add_op->AddInput(graph_builder->GetHexagonTensorId(bias_tensor_id)); in AddFullyConnectedHelper()
106 bias_add_op->AddInput(matmul_out_min); in AddFullyConnectedHelper()
107 bias_add_op->AddInput(matmul_out_max); in AddFullyConnectedHelper()
108 bias_add_op->AddInput(OpBuilder::TensorID(bias_min_const->GetID(), 0)); in AddFullyConnectedHelper()
109 bias_add_op->AddInput(OpBuilder::TensorID(bias_max_const->GetID(), 0)); in AddFullyConnectedHelper()
111 bias_add_op->AddOutput(sizeof(int), 4, in AddFullyConnectedHelper()
115 bias_add_op->AddOutput(sizeof(float), 4, scalar_shape); in AddFullyConnectedHelper()
[all …]
|
| /external/tensorflow/tensorflow/lite/toco/graph_transformations/ |
| D | identify_dilated_conv.cc | 117 Operator* bias_add_op = !has_bias_before_bts ? final_op : next_op; in ResolveDilatedConv() local
118 if (bias_add_op->type != OperatorType::kAdd) { in ResolveDilatedConv()
122 CHECK_EQ(bias_add_op->inputs.size(), 2); in ResolveDilatedConv()
123 CHECK_EQ(bias_add_op->outputs.size(), 1); in ResolveDilatedConv()
146 bias_add_op->outputs[0] = final_op->outputs[0]; in ResolveDilatedConv()
148 bias_add_op->inputs[0] = post_conv_op->outputs[0]; in ResolveDilatedConv()
153 bias_add_op->inputs[0] = conv_op->outputs[0]; in ResolveDilatedConv()
|
| /external/tensorflow/tensorflow/python/tools/ |
| D | optimize_for_inference_lib.py | 392 bias_add_op = node_def_pb2.NodeDef()
393 bias_add_op.op = "BiasAdd"
394 bias_add_op.name = node.name
395 bias_add_op.attr["T"].CopyFrom(conv_op.attr["T"])
396 bias_add_op.attr["data_format"].CopyFrom(conv_op.attr["data_format"])
397 bias_add_op.input.extend([new_conv_op.name, offset_op.name])
398 new_ops.extend([scaled_weights_op, new_conv_op, offset_op, bias_add_op])
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | fused_kernel_matcher.cc | 172 auto *bias_add_op = bias_add.getOperation(); in matchAndRewrite() local
173 if (bias_add_op) rewriter.setInsertionPoint(bias_add_op); in matchAndRewrite()
|
| /external/tensorflow/tensorflow/tools/graph_transforms/ |
| D | quantize_nodes_test.cc | 358 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()
|