91e2f847f6dd9d57af9276d8d4aeee1e1fdc9030,tensorflow_transform/saved/saved_transform_io.py,,_partially_apply_saved_transform_impl,#Any#Any#Any#Any#,66

Before Change

  // Create a map from tensor names in the graph to be imported, to the tensors
  // specified in `input_tensors`.
  input_map = {
      input_signature[decomposed_logical_name]:
      decomposed_input_tensors[decomposed_logical_name]
      for decomposed_logical_name in decomposed_input_tensors}
  input_map.update(asset_tensor_dict)
  if tensor_replacement_map:
    input_map.update(tensor_replacement_map)

After Change

  // Create a map from tensor names in the graph to be imported, to the tensors
  // specified in `input_tensors`.
  input_map = {}
  for logical_name, replacement in six.iteritems(logical_input_map):
    tensor_info = input_signature[logical_name]
    if tensor_info.WhichOneof("encoding") == "coo_sparse":
      input_map[tensor_info.coo_sparse.indices_tensor_name] = (
          replacement.indices)
      input_map[tensor_info.coo_sparse.values_tensor_name] = replacement.values
      input_map[tensor_info.coo_sparse.dense_shape_tensor_name] = (
          replacement.dense_shape)
    else:
      input_map[tensor_info.name] = replacement

  input_map.update(asset_tensor_dict)
  if tensor_replacement_map:
    input_map.update(tensor_replacement_map)

Instances