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 …]
64 variant_tensor_ops = traverse.obtain_all_variant_tensor_ops(dataset)65 remap_dict = _clone_helper(dataset._variant_tensor.op, variant_tensor_ops)74 def _clone_helper(op_to_clone, variant_tensor_ops): argument88 if input_tensor_op in variant_tensor_ops:89 recursive_map = _clone_helper(input_tensor_op, variant_tensor_ops)