Searched refs:get_op (Results 1 – 5 of 5) sorted by relevance
485 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 …]
106 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)
231 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)
186 def get_op(self, pattern_or_name): member in MatchResult
45 get_op = ged_ops.experimental_indexed_dataset_get(49 self.assertEqual([3], self.evaluate(get_op))