Searched refs:output_weights (Results 1 – 4 of 4) sorted by relevance
184 output_weights = get_dense_weights(HIDDEN_SIZE, NUM_LABELS)185 variables = hidden_weights + output_weights199 logits_act = dense_layer(output_weights, hidden_act, tf.identity)
1341 output_weights = []1344 output_weights.extend(weight.get_tensors())1346 output_weights.append(weight)1347 return backend.batch_get_value(output_weights)
1927 output_weights = []1930 output_weights.extend(weight.get_tensors())1932 output_weights.append(weight)1933 return backend.batch_get_value(output_weights)
880 TRT_ShapedWeights output_weights(nvinfer1::DataType::kFLOAT); in TEST_F() local881 params->outputs->emplace_back(output_weights); in TEST_F()