AbstractTensorHandle* AggregateGradients(
      absl::Span<AbstractTensorHandle* const> gradient_tensors) const override;

  // Calls the passed-in backward function.
  // op_type is the op's name provided in RecordOperation.
  absl::Status CallBackwardFunction(
      const string& op_type, GradientFunction* gradient_function,
      const std::vector<int64_t>& unneeded_gradients,
      absl::Span<AbstractTensorHandle* const> output_gradients,

        }
        
        public long getThroughput() {
            return bytesSent.get() + bytesReceived.get();
        }
    }
    
    public void recordOperation(String channelId, long bytes, long latency, boolean success) {
        ChannelStatistics stats = channelStats.computeIfAbsent(channelId, 
            k -> new ChannelStatistics());