Function SetDimensionSize

tensorflow/compiler/xla/client/xla_builder.cc:3661–3663

}

XlaOp SetDimensionSize(const XlaOp operand, const XlaOp val, int64 dimension) {
  return operand.builder()->SetDimensionSize(operand, val, dimension);
}

}  // namespace xla