Searched refs:input_to_ops_map (Results 1 – 3 of 3) sorted by relevance
36 input_to_ops_map = input_to_ops.InputToOps(graph)37 consumer_operations = input_to_ops_map.ConsumerOperations(input_tensor.op)47 input_to_ops_map = input_to_ops.InputToOps(graph)48 consumer_operations = input_to_ops_map.ConsumerOperations(input_tensor.op)60 input_to_ops_map = input_to_ops.InputToOps(graph)61 consumer_operations = input_to_ops_map.ConsumerOperations(input_tensor.op)
86 input_to_ops_map = input_to_ops.InputToOps(graph)99 input_to_ops_map.ConsumerOperations(layer_match.weight_tensor.op),112 consumer_ops = input_to_ops_map.ConsumerOperations(148 input_to_ops_map.ConsumerOperations(layer_match.bias_add_op),162 consumers = input_to_ops_map.ConsumerOperations(layer_match.bypass_op)171 input_to_ops_map.ConsumerOperations(layer_match.bypass_op),196 consumers = input_to_ops_map.ConsumerOperations(257 input_to_ops_map = input_to_ops.InputToOps(graph)262 consumers = input_to_ops_map.ConsumerOperations(op)
517 input_to_ops_map = input_to_ops.InputToOps(graph)520 has_scaling = _HasScaling(graph, input_to_ops_map, bn)959 def _HasScaling(graph, input_to_ops_map, bn): argument993 rsqrt_consumers = input_to_ops_map.ConsumerOperations(rsqrt_op)