Home
last modified time | relevance | path

Searched refs:fc_attr (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/lite/delegates/gpu/common/transformations/
Dmake_fully_connected.cc65 FullyConnectedAttributes fc_attr; in ApplyToNode() local
66 fc_attr.weights = conv_attr.weights; in ApplyToNode()
67 fc_attr.bias = conv_attr.bias; in ApplyToNode()
69 node->operation.attributes = fc_attr; in ApplyToNode()
Dmake_fully_connected_test.cc105 auto fc_attr = absl::any_cast<FullyConnectedAttributes>( in TEST() local
107 EXPECT_EQ(OHWI(32, 1, 1, 16), fc_attr.weights.shape); in TEST()
108 EXPECT_EQ(Linear(32), fc_attr.bias.shape); in TEST()
/external/tensorflow/tensorflow/lite/delegates/gpu/common/
Dlstm_parser.cc129 FullyConnectedAttributes fc_attr; in BuildLstmGate() local
131 SetFullyConnectedWeights(input_weight_id, reader, &fc_attr)); in BuildLstmGate()
133 RETURN_IF_ERROR(reader->ReadTensor(bias_id, &(fc_attr.bias))); in BuildLstmGate()
135 node->operation.attributes = std::move(fc_attr); in BuildLstmGate()
146 FullyConnectedAttributes fc_attr; in BuildLstmGate() local
148 SetFullyConnectedWeights(recurrent_weight_id, reader, &fc_attr)); in BuildLstmGate()
149 node->operation.attributes = std::move(fc_attr); in BuildLstmGate()
380 FullyConnectedAttributes fc_attr; in BuildOutputStateUpdate() local
383 &fc_attr)); in BuildOutputStateUpdate()
387 &(fc_attr.bias)) in BuildOutputStateUpdate()
[all …]
Dmodel_builder.cc968 FullyConnectedAttributes fc_attr; in ParseBasic() local
969 RETURN_IF_ERROR(GetFullyConnectedAttributes(2, 3, reader, &fc_attr)); in ParseBasic()
970 fc_node->operation.attributes = std::move(fc_attr); in ParseBasic()