Searched refs:variant_tensor_ops (Results 1 – 2 of 2) sorted by relevance
47 variant_tensor_ops = traverse.obtain_all_variant_tensor_ops(ds)48 self.assertAllEqual(["RangeDataset"], [x.name for x in variant_tensor_ops])53 variant_tensor_ops = traverse.obtain_all_variant_tensor_ops(ds)56 set([x.name for x in variant_tensor_ops]))63 variant_tensor_ops = traverse.obtain_all_variant_tensor_ops(ds)66 set([x.name for x in variant_tensor_ops]))73 variant_tensor_ops = traverse.obtain_all_variant_tensor_ops(ds)76 set([x.name for x in variant_tensor_ops]))82 variant_tensor_ops = traverse.obtain_all_variant_tensor_ops(ds)85 set([x.name for x in variant_tensor_ops]))[all …]
62 variant_tensor_ops = traverse.obtain_all_variant_tensor_ops(dataset)63 remap_dict = _clone_helper(dataset._variant_tensor.op, variant_tensor_ops)73 def _clone_helper(op_to_clone, variant_tensor_ops): argument87 if input_tensor_op in variant_tensor_ops:88 recursive_map = _clone_helper(input_tensor_op, variant_tensor_ops)