Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/contrib/quantize/python/
Dquantize.py485 layer_op = match_result.get_op(layer_pattern)
491 activation_op = match_result.get_op(activation_pattern)
492 bias_add_op = match_result.get_op(bias_add_pattern)
494 bias_add_op = match_result.get_op(folded_bias_add_pattern)
495 bypass_op = match_result.get_op(bypass_pattern)
496 post_activation_bypass_op = match_result.get_op(
508 layer_op = match_result.get_op(layer_pattern)
514 activation_op = match_result.get_op(activation_pattern)
515 bias_add_op = match_result.get_op(bias_add_pattern)
517 bias_add_op = match_result.get_op(folded_bias_add_pattern)
[all …]
Dgraph_matcher_test.py106 self.assertEqual(match_result.get_op(y1_pattern), y1.op)
125 match_result.get_op(add_or_sub_pattern)
154 self.assertEqual(match_result.get_op('transpose'), transpose.op)
165 self.assertEqual(match_result.get_op('transpose'), transpose.op)
188 match_result.get_op(add_pattern_a)
193 match_result.get_op(add_pattern_b)
201 match_result.get_op(add_pattern_fail)
Dfold_batch_norms.py231 layer_op = match_result.get_op(layer_pattern)
233 bn_id_op = match_result.get_op(batch_norm_identity_pattern)
234 bn_op = match_result.get_op(batch_norm_pattern)
244 output_reshape_op = match_result.get_op(matmul_bn_output_reshape_pattern)
256 batch_to_space_op = match_result.get_op(batch_to_space_pattern)
293 sub_op = mul_match_result.get_op(moving_average_sub_pattern)
Dgraph_matcher.py186 def get_op(self, pattern_or_name): member in MatchResult
/external/tensorflow/tensorflow/python/data/experimental/kernel_tests/
Dindexed_dataset_ops_test.py45 get_op = ged_ops.experimental_indexed_dataset_get(
49 self.assertEqual([3], self.evaluate(get_op))