Searched refs:variant_tensor_ops (Results 1 – 2 of 2) sorted by relevance
48 variant_tensor_ops = traverse.obtain_all_variant_tensor_ops(ds)49 self.assertAllEqual(["RangeDataset"], [x.name for x in variant_tensor_ops])54 variant_tensor_ops = traverse.obtain_all_variant_tensor_ops(ds)57 set(x.name for x in variant_tensor_ops))64 variant_tensor_ops = traverse.obtain_all_variant_tensor_ops(ds)67 set(x.name for x in variant_tensor_ops))74 variant_tensor_ops = traverse.obtain_all_variant_tensor_ops(ds)77 set(x.name for x in variant_tensor_ops))83 variant_tensor_ops = traverse.obtain_all_variant_tensor_ops(ds)86 set(x.name for x in variant_tensor_ops))[all …]
60 variant_tensor_ops = traverse.obtain_all_variant_tensor_ops(dataset)61 remap_dict = _clone_helper(dataset._variant_tensor.op, variant_tensor_ops)70 def _clone_helper(op_to_clone, variant_tensor_ops): argument84 if input_tensor_op in variant_tensor_ops:85 recursive_map = _clone_helper(input_tensor_op, variant_tensor_ops)