Searched refs:batch_to_space_op (Results 1 – 3 of 3) sorted by relevance
124 match.batch_to_space_op)230 batch_to_space_op = None256 batch_to_space_op = match_result.get_op(batch_to_space_pattern)321 batch_to_space_op=batch_to_space_op)447 batch_to_space_op): argument476 if batch_to_space_op:477 batch_to_space_op = layer_op.outputs[0].consumers()[0]480 new_batch_to_space_name = batch_to_space_op.name.split('/')[-1] + '_Fold'483 batch_to_space_op.inputs[1],484 batch_to_space_op.inputs[2],[all …]
78 Output batch_to_space_op = BatchToSpaceND( in TestFlattenAtrousConv() local
487 Output batch_to_space_op = in TestFoldFusedBatchNormsWithBatchToSpace() local